Ersetzen von numpy-Array-Elementen, die nicht Null sind

Question

Ich habe 2 numpyArrays wie:

arr1 = np.array([[0,5,5,0],[0,5,5,0]])
arr2 = np.array([[7,7,0,0],[7,7,0,0]])

Ich möchte Elemente ungleich Null an die arr2entsprechende Position kopieren, um arr1das folgende Array zu erhalten:

arr1 = np.array([[7,7,5,0],[7,7,5,0]])

Answer 1

Sie können drei Alternativen verwenden:

arr1[arr2 > 0] = arr2[arr2 > 0]
arr1[np.where(arr2>0)] = arr2[np.where(arr2>0)]
arr1[arr2.nonzero()] = arr2[arr2.nonzero()]

Aber ein Gewinner ist np.copyto(arr1, arr2, where=arr2 != 0), danke @Mark Meyer.

Jede dieser 4 Methoden ändert sich arr1in

array([[7, 7, 5, 0],
       [7, 7, 5, 0]])

Wenn Sie keine Nebenwirkungen wünschen arr1, verwenden Sie arr = arr1.copy()stattdessen und ersetzen Sie sie auf analoge Weise.

Aktualisieren

Schauen wir uns die Ergebnisse von Perfplot an

import perfplot

def simple(arr):
    arr1, arr2 = arr
    arr1[arr2 != 0] = arr2[arr2 != 0]
    return arr1
def where(arr):
    arr1, arr2 = arr
    arr1[np.where(arr2 != 0)] = arr2[np.where(arr2 != 0)]
    return arr1
def nonzero(arr):
    arr1, arr2 = arr
    arr1[arr2.nonzero()] = arr2[arr2.nonzero()]
    return arr1
def simple_improve(arr):
    arr1, arr2 = arr
    idx = arr2 != 0
    arr1[idx] = arr2[idx]
    return arr1
def where_improve(arr):
    arr1, arr2 = arr
    idx = np.where(arr2 != 0)
    arr1[idx] = arr2[idx]
    return arr1
def nonzero_improve(arr):
    arr1, arr2 = arr
    idx = arr2.nonzero()
    arr1[idx] = arr2[idx]
    return arr1
def copyto(arr): #thanks @Mark Meyer
    arr1, arr2 = arr
    np.copyto(arr1, arr2, where=arr2 != 0)
    return arr1
import numexpr as ne
def copyto_numexpr(arr):
    #some magic boost
    arr1, arr2 = arr
    np.copyto(arr1, arr2, where=ne.evaluate('arr2 != 0'))
    return arr1

perfplot.show(
    setup=lambda n: (np.tile(np.array([[0, 5, 5, 0], [0, 5, 5, 0]]), (n, n)),
                     np.tile(np.array([[7, 7, 0, 0], [7, 7, 0, 0]]), (n, n))),
    # setup=lambda n: [list(range(n))] * 10,
    kernels=[simple, where, nonzero,
             simple_improve, where_improve, nonzero_improve, 
             copyto, copyto_numexpr],
    n_range=[2 ** k for k in range(12)],
    xlabel="n*n copies of array of shape (2,4)")

Answer 2

Da Sie mutieren möchten arr1, können Sie einfach mit boolescher Indizierung zuweisen :

import numpy as np

arr1 = np.array([[0,5,5,0],[0,5,5,0]])
arr2 = np.array([[7,7,0,0],[7,7,0,0]])

arr1[arr2 != 0] = arr2[arr2 != 0]

print(arr1)

# [[7 7 5 0]
#  [7 7 5 0]]

Sie können bei großen Arrays ein wenig Leistung erzielen, indem Sie copyto():

np.copyto(arr1, arr2, where=arr2 != 0)