如何分裂混合dtype 1D numpy数组

Question

也许这个问题之前已经有答案了，但是我很难找到这个问题。假设我在一个文件中有以下数据:

date, id, int1, int2, int3
02/03/2015, 2, 23, 65, 99
10/06/2016, 4, 84, 12, 35
10/01/2017, 6, 53, 6, 78

我可以快速编写一个numpy代码片段，格式如下:

import StringIO
import numpy as np
hdr = 'date, id, int1, int2, int3'
date = '''
02/03/2015, 2, 23, 65, 99
10/06/2016, 4, 84, 12, 35
10/01/2017, 6, 53, 6, 78
'''
lines = '%s%s' % (hdr, date)
pseudo_file = StringIO.StringIO(lines)
np_dtypes = 'S10,%s' % ','.join(['i4' for x in hdr.split(',')[1:]])
np1 = np.genfromtxt(pseudo_file, delimiter=',', names=True, dtype=np_dtypes)
print np1
print np1.dtype.names
print np1.shape
print np1['date']
print np1['int3']

这将给我以下输出:

[('02/03/2015', 2, 23, 65, 99) ('10/06/2016', 4, 84, 12, 35)
 ('10/01/2017', 6, 53, 6, 78)]
('date', 'id', 'int1', 'int2', 'int3')
(3L,)
['02/03/2015' '10/06/2016' '10/01/2017']
[99 35 78]

可以看到numpy能够成功解析数组。然而，我如何将其分成两部分:

1. 仅包含字符串(日期列)的1D数组;
2. 另一个只有整数的一维数组。

拆分应该保持每个列的名称结构

Answer 1

您是否已经将字符串与np1['date']分开了?为了保留它的列名，您可以将'date'列名放在一个列表中(感谢@hpaulj):

dates=np1[['date']]
dates
#array([('02/03/2015',), ('10/06/2016',), ('10/01/2017',)], 
#      dtype=[('date', 'S10')])

要得到整型数:

ints=np1[['int1','int2','int3']]
ints
#array([(23, 65, 99), (84, 12, 35), (53, 6, 78)], 
#      dtype=[('int1', '<i4'), ('int2', '<i4'), ('int3', '<i4')])