Numpy genfromtxt列名

Question

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

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

这是一个元组，您可以按需要迭代和访问这些列。