您没有“扩展”您的 recarray X。Recarrays 不能按标签(名称/列)扩展,这是您希望使用ma.resize
. 请注意,ma.resize
返回具有新形状的新(屏蔽)数组而不更改传递给它的数组,但在您的代码中您没有使用返回值。所以那条线没有做任何事情。澄清:
X[afield] = ma.resize(X[afield], (len(Y),) )
也不起作用,因为记录数组不能按标签(“列”)扩展。如果你想扩展一个recarray,你需要一次性完成(使用函数 from np.lib.recfunctions
),所以添加一个全新的列或为所有现有列添加几条新记录。
话虽如此,为什么不试试这个:
>>> Y = np.arange(20, dtype=np.float)
>>> Ynan = np.insert(Y, (5,)*3, (np.nan,)*3)
>>> X = np.rec.fromarrays([Ynan, Ynan+100], names='g,h')
>>> X
rec.array([(0.0, 100.0), (1.0, 101.0), (2.0, 102.0), (3.0, 103.0),
(4.0, 104.0), (nan, nan), (nan, nan), (nan, nan), (5.0, 105.0),
(6.0, 106.0), (7.0, 107.0), (8.0, 108.0), (9.0, 109.0),
(10.0, 110.0), (11.0, 111.0), (12.0, 112.0), (13.0, 113.0),
(14.0, 114.0), (15.0, 115.0), (16.0, 116.0), (17.0, 117.0),
(18.0, 118.0), (19.0, 119.0)],
dtype=[('g', '<f8'), ('h', '<f8')])
请注意,您不能将第二列(标签“h”)转换为 int,因为np.nan
它是浮点类型。如果你尝试过,你会得到垃圾:
>>> X['h'].astype(np.int)
array([ 100, 101, 102,
103, 104, -9223372036854775808,
-9223372036854775808, -9223372036854775808, 105,
106, 107, 108,
109, 110, 111,
112, 113, 114,
115, 116, 117,
118, 119])
我认为您所追求的实际上是蒙面的记录数组:
>>> import numpy.ma.mrecords as mrecords
>>>
>>> X = np.rec.fromarrays([Ynan, (Ynan+100).astype(np.int)], names='g,h')
>>> Z = np.ma.array(X, mask=np.isnan(Ynan))
>>> Z2 = Z.view(mrecords.mrecarray)
>>>
>>> Z2
masked_records(
g : [0.0 1.0 2.0 3.0 4.0 -- -- -- 5.0 6.0 7.0 8.0 9.0 10.0 11.0 12.0 13.0 14.0
15.0 16.0 17.0 18.0 19.0]
h : [100 101 102 103 104 -- -- -- 105 106 107 108 109 110 111 112 113 114 115
116 117 118 119]
fill_value : (1e+20, 999999)
)
>>>
>>> Z2['h']
masked_array(data = [100 101 102 103 104 -- -- -- 105 106 107 108 109 110 111 112 113 114 115
116 117 118 119],
mask = [False False False False False True True True False False False False
False False False False False False False False False False False],
fill_value = 999999)
如您所见,Z2 的“列”具有所需的 dtype(float 和 int),可通过列名访问,并屏蔽了一些数据。