python - 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

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