根据条件合并数组

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中取值在C中，如果A中的值大于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]

Answer 2

np.where()可能是首选解决方案，如 @Ghilas BELHADJ 所述。如果你想要列表理解版本（可能没有numpy使用）

max_c = np.max(C)
D = np.array([A[i] if A[i] < max_c else B[i] for i in range(len(A))])