添加2个Numpy掩盖阵列时的奇怪错误

Question

因此，我正在尝试将2个Numpy掩盖阵列添加在一起。困难是必须将它们添加为字符串，因为IM试图在产生的输出数组中获取二进制代码。下面的代码是我要做的事情的简化版本。两个阵列的掩码都是相同的（实际上，这些阵列会更大，但想法是相同的）：

a = np.zeros((3,3))
b = np.ones((3,3))
amask = [[False,True,True],[True, True, False],[False, False , True]]
bmask = [[False,True,True],[True, True, False],[False, False , True]]

a = a.astype('str')
b= b.astype('str')

am = ma.masked_array(a,mask = amask)
bm = ma.masked_array(b, mask = bmask)
x = np.add(am,bm)

我希望输出成为类似的东西：

[['01' -- --],[-- -- '01'],['01', '01' --]]

因此，它对于它来说是非常重要的，因此它们可以这样添加。

但是，运行此代码会给我以下错误：

numpy.core._exceptions.UFuncTypeError: ufunc 'add' did not contain a loop with signature matching types (dtype('<U32'), dtype('<U32')) -> None

我认为这不理解，因为我认为两个数组显然具有相同的数据类型。在没有字符串转换的情况下添加它们效果很好，但不会给我所需的输出。我以前遇到过这个错误，并尝试查找它，但从未真正理解过。谢谢

Answer 1

这不是蒙版的阵列问题；这是一个字符串dtype。

In [254]: a = np.arange(4)
In [255]: a
Out[255]: array([0, 1, 2, 3])
In [256]: a+a
Out[256]: array([0, 2, 4, 6])
In [257]: a1 = a.astype(str)
In [258]: a1
Out[258]: array(['0', '1', '2', '3'], dtype='<U21')
In [259]: a1 + a1
Traceback (most recent call last):
  Input In [259] in <cell line: 1>
    a1 + a1
UFuncTypeError: ufunc 'add' did not contain a loop with signature matching types (dtype('<U21'), dtype('<U21')) -> None

astype（str）用numpy string dtype制作一个数组；这是用于数组存储的优化，但与Python字符串不同。 np.char具有一些可以将字符串方法应用于 dtypes：

In [260]: np.char.add(a1,a1)
Out[260]: array(['00', '11', '22', '33'], dtype='<U42')

或如注释时，您可以列出字符串对象数组的列表：

In [261]: a2 = a1.astype(object)
In [262]: a2
Out[262]: array(['0', '1', '2', '3'], dtype=object)
In [263]: a2 + a2
Out[263]: array(['00', '11', '22', '33'], dtype=object)

对于对象dtype数组，操作员，像+将操作委派给元素方法。等效：

In [264]: [i+j for i,j in zip(a2,a2)]
Out[264]: ['00', '11', '22', '33']

我希望[264]最快。 numpy不会为字符串处理添加太多。