python - Numpy：无法读取数据集标头

Question

我的代码看起来像

import numpy as np
from numpy import genfromtxt

train = genfromtxt('/Users/hhimanshu/Downloads/dataset/digitrecognizer/train.csv',
                   delimiter=',', names=True)

并且train.csv有类似的数据

label,pixel0,pixel1,pixel2,pixel3, .....
1,0,0,0, ...

当我做

train.names

我明白了

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-62-dea84e8fe1f5> in <module>()
      4 print 'size(Number of elements in array) = ', train.size
      5 # print 'data type = ', train.dtype
----> 6 train.names

AttributeError: 'numpy.ndarray' object has no attribute 'names'

形状对我来说也不正确

shape(Tuple of array dimensions) =  (42000,)
dimension(Number of array dimensions) =  1
size(Number of elements in array) =  42000

我知道我有785列和2 个维度（这里说 1）

我做错了什么？

Answer 1

我的错，

ndarray没有名为names 我可以引用列train['labels']并获取数据的属性

Answer 2

要获取列名，您可以使用：train.dtype.names

这仅适用train于记录（结构化）数组。