numpy中掩码数组错误

Question

我用 genfromtxt 输入了一个文件，但里面有些值是缺失的，所以我生成了一个被屏蔽的数组。当我尝试索引这个被屏蔽数组中的某些记录值时，出现了一个我搞不懂的错误。希望能得到一些帮助。谢谢。 --Alex

import csv
import datetime
import time
import numpy as np
import numpy.lib.recfunctions as rf
import pprint
import numpy.ma as ma

date_converter = lambda x: datetime.date(int(x[-4:]), int(x[3:5]), int(x[:2]))
input_file = np.genfromtxt("../data/test.csv", usemask=True, converters={0:date_converter}, dtype="O4, i8, i8, i8, i8", names="date, firm, val1, val2, val3", delimiter=",", skip_header=1)

生成了：

masked_array(data = [(datetime.date(2001, 3, 1), 1L, --, 14L, 15L)
 (datetime.date(2001, 2, 1), 1L, 10L, 11L, 12L)
 (datetime.date(2001, 5, 1), 1L, 19L, 20L, 21L)
 (datetime.date(2001, 4, 1), 1L, 16L, --, 18L)],
             mask = [(False, False, True, False, False) (False, False, False, False, False)
 (False, False, False, False, False) (False, False, False, True, False)],
       fill_value = ('?', 999999L, 999999L, 999999L, 999999L),
            dtype = [('date', '|O4'), ('firm', '<i8'), ('val1', '<i8'), ('val2', '<i8'), ('val3', '<i8')])

当我运行 input_file[0] 时，出现了以下错误：

Traceback (most recent call last):
  File "<pyshell#278>", line 1, in <module>
    input_file[0]
  File "C:\Python27\lib\site-packages\numpy\ma\core.py", line 2956, in __getitem__
    dout = mvoid(dout, mask=mask)
  File "C:\Python27\lib\site-packages\numpy\ma\core.py", line 5529, in __new__
    _data[()] = data
ValueError: Setting void-array with object members using buffer.

Answer 1

input_file[0] 这种方式并不适合用来访问掩码数组中的数据（具体可以查看文档）

比如说：

>>> import numpy as np
>>> arr = np.ma.ones(3, dtype=[('c1', np.int),('c2', np.int)])
>>> arr.mask[0][1] = True
>>> arr.data[0][0] = 2              
>>> np.ma.getdata(arr)[1][0] = 3    
>>> arr.data[2][0] = 4       
>>> print(arr)
   [(2, --) (3, 1) (4, 1)]