如何获取dtype列名称

时间:2013-09-26 01:34:44

标签: python numpy

假设我使用genfromtxt创建<type 'numpy.ndarray'>

data = np.genfromtxt("test.txt",dtype=None,delimiter=',',names=True)

这导致:

array((500, 501, 502, 503, 504, 504, 504), dtype=[('ColumnName1', '<i8'), ('ColumnName2', '<i8'), ('ColumnName3', '<i8'), ('ColumnName4', '<i8'), ('ColumnName5', '<i8'), ('ColumnName6', '<i8'), ('ColumnName7', '<i8')])

我想知道的是如何检索列名? data.dtype似乎没有让我到那里。

2 个答案:

答案 0 :(得分:2)

data.dtype 完全你应该在哪里寻找。还有哪里?你想要:

data.dtype.names

答案 1 :(得分:2)

这似乎有效,至于data.dtype.names

的结果
>>> import numpy as np
>>> data=np.array((500, 501, 502, 503, 504, 504, 504), 
    dtype=[('ColumnName1', '<i8'), ('ColumnName2', '<i8'), 
    ('ColumnName3', '<i8'), ('ColumnName4', '<i8'), ('ColumnName5', '<i8'),
    ('ColumnName6', '<i8'), ('ColumnName7', '<i8')])

给出

>>> data.dtype.names
('ColumnName1', 'ColumnName2', 'ColumnName3', 
   'ColumnName4', 'ColumnName5', 'ColumnName6', 'ColumnName7')