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

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