Python中使用genfromtxt拆分列赋值变量时解包错误求助

错误原因 #

你使用np.genfromtxt(..., unpack=True)时，6列数据会被拆分成6个独立的一维数组，但你只声明了3个变量（mCH, tb, BR）来接收，变量数量与拆分出的数组数量不匹配，因此触发报错。

两种可行解决方案 #

方案1：先读取完整数据再拆分（直观易懂）

先将所有数据读取为二维数组，再通过索引提取指定列：

import numpy as np

# 读取完整数据，得到形状为(N,6)的二维数组
data = np.genfromtxt(BRfile)
# 提取第1列赋值给mCH，第2列赋值给tb
mCH = data[:, 0]
tb = data[:, 1]
# 提取第3到第6列（索引2到5）赋值给BR，形状为(N,4)
BR = data[:, 2:]

# 循环绘图时，取BR的第i列（[:,i]）
for i in range(4):
    CS = axs[i].tricontour(mCH, tb, BR[:, i], cmap='gnuplot')

方案2：用*接收剩余列（更简洁）

利用Python可变参数语法*，直接将前两列赋值给mCH、tb，剩余4列自动打包为列表BR：

import numpy as np

# *BR会把剩余4列的数组打包成列表
mCH, tb, *BR = np.genfromtxt(BRfile, unpack=True)

# 此时BR是包含4个一维数组的列表，直接用BR[i]取第i列即可
for i in range(4):
    CS = axs[i].tricontour(mCH, tb, BR[i], cmap='gnuplot')

关键说明 #

• 方案1中BR是二维数组，取对应列需要用[:,i]；方案2中BR是一维数组的列表，直接用BR[i]就能拿到目标列数据。
• 两种方案均可解决变量拆分问题，可根据习惯选择。

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