numpy.genfromtxt（）可以返回列数据列表，而不是行数据吗？

Question

我有一个类似

的txt文件

col1 col2 col3
1    4    7
2    5    8
3    6    9

我正在尝试生成包含列数据的列表列表：

[ [1,2,3] [4,5,6] [7,8,9] ]

但以下代码生成行数据列表，即[ [1,4,7], ... ]。

blah = np.genfromtxt(scrubbed_file, skip_header=1)
for bluh in blah:
    print(bluh)

有没有简单的方法来实现这一目标？

Answer 1

你可以像这样转置结果数组： blah = np.genfromtxt(scrubbed_file, skip_header=1).T 这将有效地将阵列置于其一侧，从而产生您想要的数据格式。

Answer 2

使用genfromtext中的解包变量：

np.genfromtxt('text.csv', delimiter=',', unpack=True, skip_header=1)

它提供以下内容：

array([[ 1.,  2.,  3.],
       [ 4.,  5.,  6.],
       [ 7.,  8.,  9.]])

Answer 3

有一个非常方便的python功能就是这样做，即zip(*list)：

blah = np.genfromtxt(scrubbed_file, skip_header=1)
for bluh in zip(*blah):
    print(bluh)