Numpy broadcast_to für maskierte Array

Question

Ich verwende die np.broadcast_to Funktion eine Ansicht auf ein neu geformtes Array zu erhalten wie das Beispiel:

>>> x = np.array([1, 2, 3]) 
>>> np.broadcast_to(x, (3, 3)) 
array([[1, 2, 3], 
     [1, 2, 3], 
     [1, 2, 3]]) 

eine maskierte Array an diese Funktion Passing verliert mir die Maske aber:

>>> y = np.ma.array([1, 2, 3], mask=[False, True, False]) 
>>> np.broadcast_to(y, (3, 3)) 
array([[1, 2, 3], 
     [1, 2, 3], 
     [1, 2, 3]]) 

Wie bekomme ich die folgende Ansicht?

array([[1, --, 3], 
     [1, --, 3], 
     [1, --, 3]]) 

Answer 1

Anscheinend können Sie einen subok Parameter np.broadcast_to passieren den Typ des übergebenen Array zu behalten und nicht Verwenden Sie den Basis-Array-Typ, aber dies überträgt nur die Daten des maskierten Arrays, nicht die Maske.

Sie sollten wahrscheinlich die Maske anschließend manuell übertragen:

>>> y = np.ma.array([1, 2, 3], mask=[False, True, False]) 
>>> z = np.broadcast_to(y, (3, 3), subok=True) 
>>> z.mask 
False 
>>> z.mask = np.broadcast_to(y.mask, z.shape) 
>>> z 
masked_array(data = 
[[1 -- 3] 
[1 -- 3] 
[1 -- 3]], 
      mask = 
[[False True False] 
[False True False] 
[False True False]], 
     fill_value = 999999) 

Answer 2

Ich denke, das ist was du willst. Maskiere das Array nach dem Broadcast, damit du das gewünschte maskierte Array bekommst.

y = np.ma.array([1, 2, 3]) 
z = np.broadcast_to(y, (3, 3)) 
x = np.ma.array(z, mask=np.broadcast_to([False,True,False], (3, 3))) 
x 
masked_array(data = 
[[1 -- 3] 
[1 -- 3] 
[1 -- 3]], 
      mask = 
[[False True False] 
[False True False] 
[False True False]], 
     fill_value = 999999) 

Überprüfen Sie, ob dies für Ihren Fall funktioniert. Wenn Sie die maskierten Array-Werte ohne wollen ‚-‘

x.compressed() 
array([1, 3, 1, 3, 1, 3]) 

Für weitere Informationen gehen Sie durch Masked array documentation