python - 根据条件合并数组

Question

我有三个 numpy 数组:

A = np.array([41, 162, 349, 641, 531, 445, 35])
B = np.array([42, 162, 323, 479, 436, 389, 36])
C = np.array([1.4, 7, 14, 28, 70, 140])

我想生成第四个数组 D，其维度与 A 和 B 相同，如果 A 中的值小于最大值，则将 A 中的每个值与 C 中的最大值进行比较，取 A 中的值如果 A 中的值大于 C 中的最大值，则 C 中的值和 B 中的值。因此在这种情况下:

np.max(C) = 140

因此 D 将是:

D = np.array([41, 162, 323, 479, 436, 389, 35])

Answer 1

np.where()就是为此而生的。

D = np.where(A<C.max(), A, B)
# [ 41 162 323 479 436 389  35]