python - ndarray 可以存储日期时间和 float 吗？

Question

如何制作这样一个可以同时存储日期时间和 float 的 numpy 数据结构？

array([[ 2017-01-30 00:00:00,  1.0],
       [ 2017-01-31 00:00:00,  2.0]])

Answer 1

您可以使用 structured array具有异质元组:

import numpy as np
x = np.array([(np.datetime64('2017-01-30'), 1.0),
              (np.datetime64('2017-01-31'), 2.0)],
              dtype=[('datetime', 'datetime64[D]'), ('number', 'f8')])

语法有点类似于dicts then:

>>> x['datetime']
array(['2017-01-30', '2017-01-31'], dtype='datetime64[D]')
>>> x['number']
array([ 1.,  2.])
>>> x['datetime'][0] + 5
numpy.datetime64('2017-02-04')
>>> x['number'][1] + 5
7.0

请注意 Pandas可能更适合您的需求。