如何将二维复数Numpy数组导出为指定格式的dat文件？

要将你生成的二维复数NumPy数组导出为指定格式的dat文件，可按以下步骤操作：

1. 生成匹配的x、y一维序列：
   原数组按indexing='ij'生成，每个x值对应所有y值，因此通过重复x数组、平铺y数组，得到与数据行一一对应的x、y序列。

2. 提取复数数组的实部和虚部并展平：
   用z.real和z.imag获取实部、虚部数组，再展平为一维，与x、y序列匹配。

3. 组合数据并写入文件：
   将x、y、实部、虚部组合成二维数组，按指定格式写入文件，确保表头和数据对齐。

完整代码示例 #

import numpy as np

# 原数组生成代码
xl = np.linspace(0, 2, num=3)
yl = np.linspace(0, 1, num=2)
xm, ym = np.meshgrid(xl, yl, sparse=True, indexing='ij')
z = xm**2 + 1j*ym

# 生成匹配的x、y一维序列
x_flat = xl.repeat(len(yl))  # 每个x值重复y的长度次：[0,0,1,1,2,2]
y_flat = np.tile(yl, len(xl))  # 平铺y序列：[0,1,0,1,0,1]

# 提取实部和虚部并展平
z_real = z.real.flatten()
z_imag = z.imag.flatten()

# 组合成二维数据数组
data = np.column_stack((x_flat, y_flat, z_real, z_imag))

# 写入dat文件
with open('output.dat', 'w') as f:
    # 写入表头
    f.write('  x    y    Re(z)   Im(z)\n')
    # 按格式写入每行数据，确保对齐
    for row in data:
        # 若数据为浮点数，可将int()和:d改为%.2f等格式
        f.write(f'{int(row[0]):4d}{int(row[1]):4d}{int(row[2]):8d}{int(row[3]):8d}\n')

补充说明 #

• 如果你的x、y或z的实部/虚部是浮点数，只需调整格式字符串，比如将int(row[0]):4d改为row[0]:4.2f，即可保留小数位。
• 也可使用np.savetxt简化写入操作：

np.savetxt('output.dat', data, fmt='%4d %4d %8d %8d', header='  x    y    Re(z)   Im(z)', comments='')

设置comments=''是为了避免表头被自动添加#注释符号。

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