python - 存储数据以与 Numpy 一起使用，带有自动标题

Question

我有一个具有这种设计的巨大数据文件:

A B C D
2 3 1 2
3 1 3 6
1 5 4 1

我想将其存储在 numpy 数组中，因为我需要对它们进行大量计算。所以我想要这样的数组，其中数组的名称是自动创建的:

A=[2,3,1]
B=[3,1,5]
C=[1,3,4]
D=[2,6,1]

是否可以为这些数组自动生成名称？

Answer 1

您可以创建一个 numpy 结构化数组，其中每一列的名称取自输入的第一行，使用 genfromtxt并将 names 设置为 True

import io
import numpy as np

t=u"""A B C D
      2 3 1 2
      3 1 3 6
      1 5 4 1"""

data = np.genfromtxt(io.StringIO(t),names=True)

print data['A']
# [ 2.,  3.,  1.]

print data['B']
# [ 3.,  1.,  5.]