python numpy结构化数组问题

Question

我对 numpy 比较陌生。我已从 .csv 文件导入数据，日期格式为 YYYY、MM、DD 和其他一些内容。我想将所有内容放入一个数组中，并使用“正确”日期时间格式的日期。这是我的代码:

na_trades = np.zeros((number_of_orders,), dtype = ('datetime64,a5,a5,i4'))
for row in range(number_of_orders):
    order = na_trades_csv[row]
    order_date = dt.datetime(order[0],order[1],order[2])
    order_date64 =  np.datetime64(order_date)
    na_trades[row] = (order_date64,order[3],order[4],order[5])

但是我收到错误ValueError:使用序列设置数组元素时出错。知道这是为什么吗？提前感谢您的帮助!

Answer 1

使用 numpy 版本 1.6.2，dtype = 'datetime64,a5,a5,i4' 不会产生预期的 dtype:

In [36]: na_trades = np.zeros((number_of_orders,), dtype = 'datetime64,a5,a5,i4')
In [37]: na_trades
Out[37]: array([1970-01-01 00:00:00], dtype=datetime64[us])

这对我来说看起来像是一个错误——尽管我可能是错的。尝试一下:

na_trades = np.empty(number_of_orders,
                     dtype = [
                         ('dt', 'datetime64'),
                         ('foo','a5'),
                         ('bar', 'a5'),
                         ('baz', 'i4')])