Numpy genfromtxt列名

Question

如何让genfromtxt返回我的列名list ，这些列名是由names=True自动检索的？ 当我做：

data = np.genfromtxt("test.csv",names=True,delimiter=",",dtype=None)
print data['col1']

它打印col1的整个列值。

但是，我需要遍历所有列名称。 我怎样才能做到这一点？

我尝试了data.keys()和其他各种方法，但genfromtxt返回的任何内容似乎都不是字典兼容的对象。 我想我可以自己传递列名list ，但从长远来看这对我来说是不可维护的。

有任何想法吗？

Answer 1

genfromtxt返回一个numpy.ndarray 。

您可以使用获取数据类型

data.dtype

或者只是带有的名字

data.dtype.names

这是一个元组，您可以迭代并根据需要访问列。