将SGY文件转为Numpy数组并保存为CSV时遇TypeError求助

错误根源 #

触发TypeError: can only concatenate str (not "list") to str的直接原因是调用函数时的参数错误：

• data_dir_SGY是字符串路径，names_full是文件名列表，直接执行data_dir_SGY + names_full属于试图把字符串和列表拼接，Python不允许这种操作，直接触发类型错误。

除此之外，代码还有其他逻辑问题：

1. 循环条件错误：range(len(names_full[i]) -1)里的names_full[i]是单个字符串，len()取的是字符串长度而非列表长度，循环逻辑完全错误。
2. the_array未初始化：直接用the_array[i]赋值会触发NameError。
3. 依赖全局变量：names_full、names_no_format、data_dir_CSV都是全局变量，函数耦合性太高，易出错。
4. 模块导入放在函数内部：虽然能运行，但不符合Python规范，影响性能。

修正后的代码 #

import numpy as np
from obspy.io.segy.core import _read_segy

def segy_2_csv(data_dir_sgy, names_full, names_no_format, data_dir_csv):
    # 遍历所有文件名索引
    for i in range(len(names_full)):
        # 拼接完整的SEGY文件路径
        segy_path = f"{data_dir_sgy}/{names_full[i]}"
        # 读取SEGY文件并转换为二维数组
        segy_obj = _read_segy(segy_path)
        segy_array = np.array(segy_obj).transpose((1, 0))
        # 拼接CSV保存路径
        csv_path = f"{data_dir_csv}/{names_no_format[i]}.csv"
        # 保存为CSV文件
        np.savetxt(csv_path, segy_array, delimiter=",")

# 调用示例（替换成你的实际路径和文件名列表）
data_dir_SGY = "/path/to/your/segy/files"
data_dir_CSV = "/path/to/save/csv/files"
names_full = ["file1.sgy", "file2.sgy"]
names_no_format = ["file1", "file2"]

segy_2_csv(data_dir_SGY, names_full, names_no_format, data_dir_CSV)

关键修改说明 #

• 参数传递优化：把所有需要的路径和文件名列表作为参数传入函数，不再依赖全局变量，逻辑更清晰，复用性更强。
• 路径拼接修正：用f-string拼接完整文件路径，避免字符串与列表拼接的错误，同时保证路径格式正确（自动处理分隔符）。
• 循环逻辑修正：改为遍历names_full的长度，确保每个SEGY文件都被处理。
• 移除无用数组：直接读取后保存，不需要the_array存储，节省内存。
• 模块移到外部：将numpy和obspy的导入放在函数外，符合Python编码规范。

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