1

我正在尝试使用现有数组的元素创建一个新数组。我不断收到:ValueError: Setting void-array with object members using buffer。

import numpy as np
import datetime

date = datetime.date(2013, 4, 5)
results = [(date,0,1,2,3), (date,5,1,5,6), (date,3,4,4,7)] 
stock_dt = np.dtype([('date', object),
                     ('open', np.int8),
                     ('high', np.int8),
                     ('low', np.int8),
                     ('close', np.int8)])

d = np.array(results, dtype=stock_dt)
matches = []
for item in d:
    if item['high'] == 1:
        matches.append(item)

rec = np.array(matches, dtype=stock_dt)

print rec
4

2 回答 2

4

问题是它matches不是一个元组列表,所以你不能用它制作一个结构化数组。相反,它是结构化数组的列表,需要合并回单个结构化数组。您可以numpy.lib.recfunctions.stack_arrays为此使用:

In [21]: import numpy.lib.recfunctions as rfn

In [22]: rfn.stack_arrays(matches,usemask=False)
Out[22]: 
array([(datetime.date(2013, 4, 5), 0, 1, 2, 3),
       (datetime.date(2013, 4, 5), 5, 1, 5, 6)], 
      dtype=[('date', 'O'), ('open', 'i1'), ('high', 'i1'), ('low', 'i1'), ('close', 'i1')])

您还可以考虑完全取消循环:

In [23]: d[d['high'] == 1]
Out[23]: 
array([(datetime.date(2013, 4, 5), 0, 1, 2, 3),
       (datetime.date(2013, 4, 5), 5, 1, 5, 6)], 
      dtype=[('date', 'O'), ('open', 'i1'), ('high', 'i1'), ('low', 'i1'), ('close', 'i1')])

哪个应该更快,启动。

于 2013-03-28T02:40:17.347 回答
1

改变

rec = np.array(matches, dtype=stock_dt)

rec = np.array(matches)

当您进行迭代时,matches您不再处理元组,因此您不应该再次传递dtype=stock_dtnp.array

于 2013-03-28T02:36:25.410 回答