Python - 根據標准組合兩個 arrays

Question

我正在嘗試根據某些標准將兩個 numpy 2D arrays 組合成一個數組，例如

# array1
[1,2,3,4,5]
[10,11,12,13,14]

# array2
[15,16,17,18,19]
[6,7,8,9,10] 

根據標准 - 最高值 - 生成的數組 1 和數組 2 的組合如下所示；

[15,16,17,18,19]
[10,11,12,13,14]

我知道如何將兩個 arrays 連接在一起，過濾，但我似乎找不到可以為我提供上述結果的 function。

也許是np.where的變體？

Answer 1

使用numpy.where ，對於 arrays a和b ，您可以使用：

np.where(a>b, a, b)

對於您的示例：

>>> import numpy as np
>>> a = np.array([[1,2,3,4,5],[10,11,12,13,14]])
>>> b = np.array([[15,16,17,18,19],[6,7,8,9,10]])
>>> np.where(a>b, a, b)
array([[15, 16, 17, 18, 19],
       [10, 11, 12, 13, 14]])

Answer 2

您可以使用numpy.maximum ：

import numpy as np

a1 = np.array([[1,2,3,4,5],[10,11,12,13,14]])
a2 = np.array([[15,16,17,18,19],[10,11,12,13,14]])

print(np.maximum(a1, a2))
# [[15 16 17 18 19]
#  [10 11 12 13 14]]