Python读取含复数元素的data.OUT文件的解决方案求助

numpy的genfromtxt和loadtxt默认按空格分割字段，但你的data.OUT文件中后两列是方括号包裹的复数列表格式（如[(-5.254878050808644e-16+0j), (1.9349052915745023e-16+0j)]），这种格式无法被numpy的complex dtype直接解析，因此会出现返回nan+0j或报错的情况。

方案1：逐行手动解析文本（最直接） #

逐行读取文件，分割字段后分别处理数值和复数列表，最后合并为numpy数组：

import numpy as np

data = []
with open("data.OUT", "r") as f:
    for line in f:
        # 自动处理多空格分割
        segments = line.strip().split()
        # 前7个字段转float
        numeric_part = list(map(float, segments[:7]))
        # 处理第8个复数列表
        comp_list1 = segments[7].strip("[]").split("), (")
        comps1 = [complex(c.strip("()")) for c in comp_list1]
        # 处理第9个复数列表
        comp_list2 = segments[8].strip("[]").split("), (")
        comps2 = [complex(c.strip("()")) for c in comp_list2]
        # 合并当前行所有数据
        data.append(numeric_part + comps1 + comps2)

# 转换为numpy复数数组
result = np.array(data, dtype=complex)
print(result)

方案2：使用genfromtxt配合自定义转换器 #

利用numpy的转换器功能，针对复数列表字段编写解析函数，实现批量读取：

import numpy as np

def parse_complex_list(byte_str):
    # 转换字节串为字符串，去除首尾方括号
    str_content = byte_str.decode().strip("[]")
    # 分割每个复数字符串
    comp_strings = str_content.split("), (")
    # 逐个转换为复数
    return [complex(s.strip("()")) for s in comp_strings]

# 定义各列转换器：前7列用float，后两列用自定义解析函数
converters = {i: float for i in range(7)}
converters[7] = parse_complex_list
converters[8] = parse_complex_list

# 读取数据，dtype设为object以兼容列表类型
raw_data = np.genfromtxt("data.OUT", converters=converters, dtype=object)

# 可选：将列表展开为独立列，转换为纯复数数组
expanded_data = []
for row in raw_data:
    expanded_row = list(row[:7]) + row[7] + row[8]
    expanded_data.append(expanded_row)
final_data = np.array(expanded_data, dtype=complex)
print(final_data)

说明 #

• 如果需要保留复数的列表结构（而非展开为单独列），可以在方案1中直接将numeric_part + [comps1, comps2]加入data，最终数组的dtype设为object。
• 两种方案都能正确解析文件中的复数格式，避免numpy默认解析的错误。

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