使用Flopy提取CBB文件数据异常：输出TXT文件为空求助

问题根源 #

你的代码存在两个核心问题：

1. 未指定目标记录类型：cbb.get_data(totim=time)默认返回所有类型的单元格预算记录，若不指定你需要的条目（比如BASEFLOW），可能返回空列表或非目标数据结构。
2. 数据遍历逻辑错误：CBB文件的单条记录是结构化数组，包含k（层）、i（行）、j（列）、q（流量）等字段，你的多层嵌套循环因结构不匹配，会跳过数据写入步骤。

修正后的代码 #

import flopy

cbb_file = r'D:\ZFZ\TBW_Project\IHM_Sim\IHM\IHM_input\Calibration\MODFLOW\gv37.cbb'
output_file_baseflow = r'D:\ZFZ\TBW_Project\IHM_Sim\IHM\IHM_input\Calibration\MODFLOW\baseflow.txt'  

# 打开CBB文件
cbb = flopy.utils.CellBudgetFile(cbb_file)

# 打印所有可用的文本条目，确认目标项存在
print("可用的单元格预算条目：", cbb.list_unique_records())

# 指定要提取的记录类型（替换为你实际需要的条目，比如'BASEFLOW'）
target_record = 'BASEFLOW'

with open(output_file_baseflow, 'w') as file:
    file.write("Text Entry, Time, Layer, Row, Column, Flow Data\n")
    
    # 遍历所有时间步
    for time in cbb.get_times():
        # 获取指定类型的记录，仅处理目标数据
        baseflow_records = cbb.get_data(totim=time, text=target_record)
        
        # 当前时间步无目标记录则跳过
        if not baseflow_records:
            continue
        
        # 遍历每个层的结构化数组记录
        for record in baseflow_records:
            # 直接遍历结构化数组的每一行数据
            for entry in record:
                # Flopy索引通常从1开始，若需0索引可自行减1
                file.write(f"{target_record}, {time}, {entry['k']}, {entry['i']}, {entry['j']}, {entry['q']}\n")

print(f"流量数据已保存到 {output_file_baseflow}")

关键修正点 #

• 确认记录类型：通过cbb.list_unique_records()先确认CBB文件中存在你要提取的条目，避免提取不存在的记录导致空输出。
• 明确指定记录类型：调用get_data()时传入text=target_record，仅获取目标类型的流量数据。
• 适配结构化数组遍历：直接遍历CBB记录的结构化数组条目，提取k,i,j,q字段，无需多层嵌套循环。
• 空记录跳过：增加空记录判断，避免因无数据引发的报错或无效循环。

内容的提问来源于stack exchange，提问作者Fan