将值(从0除外)从数组2复制到数组1

Question

我有2个具有相同形状的numpy阵列.现在我想将除0之外的所有值从数组2复制到数组1.

数组1:

[1, 1, 1]
[1, 1, 1]
[1, 1, 1]

数组2:

[0, 2, 0]
[4, 0, 0]
[6, 6, 0]

结果现在应该如下所示:

[1, 2, 1]
[4, 1, 1]
[6, 6, 1]

这在Python中怎么可能？

Answer 1

nonzero 将返回不为0的数组的索引.

idx_nonzero = B.nonzero()
A[idx_nonzero] = B[idx_nonzero]

nonzero也什么numpy.where时候只有一个条件被传递返回.因此,等价地,我们可以做

idx_nonzero = np.where(B != 0)  # (B != 0).nonzero()
A[idx_nonzero] = B[idx_nonzero]

该解决方案就位.如果您需要创建一个新数组,请参阅@jp_data_analysis'的答案.

Answer 2

np.where支持这个.下面的解决方案创建一个新数组 对于就地替代方案,请参阅@ Tai的答案.

A = np.array(
[[1, 1, 1],
[1, 1, 1],
[1, 1, 1]])

B = np.array(
[[0, 2, 0],
[4, 0, 0],
[6, 6, 0]])

C = np.where(B==0, A, B)

# [1, 2, 1]
# [4, 1, 1]
# [6, 6, 1]