python - NumPy 中的元素数组最大值函数(两个以上的数组)

Question

我试图在逐元素比较中返回多个数组的最大值。例如:

A = array([0, 1, 2])
B = array([1, 0, 3])
C = array([3, 0, 4])

我希望得到的数组是 array([3,1,4]).

我想使用 numpy.maximum，但它只适用于两个数组。有两个以上数组的简单函数吗？

Answer 1

使用此设置:

>>> A = np.array([0,1,2])
>>> B = np.array([1,0,3])
>>> C = np.array([3,0,4])

你可以这样做:

>>> np.maximum.reduce([A,B,C])
array([3, 1, 4])

或者:

>>> np.vstack([A,B,C]).max(axis=0)
array([3, 1, 4])

我会选择第一个选项。