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

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