多个numpy数组的最大值

Maximum of multiple numpy arrays

我们可以计算 3 个 numpy 数组的元素最大值:

import numpy as np
A = np.arange(20).reshape((4, 5))       # any 4x5 array
B = np.maximum(A, A+7, A+2)     # working

但为什么 np.maximum 不接受来自“解包”的多个数组?

L = [np.roll(A, k, axis=0) for k in range(4)]     # 4 arrays: A shifted with different k
np.maximum(*L)

错误:

ValueError: invalid number of arguments

毕竟,L 是一个 Python Numpy 数组对象列表,因此 *L 应该为 np.maximum 函数调用解压它。为什么它不起作用?

PS:我也尝试过 L = (...)(它提供了一个生成器)或 L = tuple(...),但我们有同样的错误。

正如评论中指出的那样,reduce 是这里的解决方案:

np.maximum.reduce([np.roll(A, k, axis=0) for k in range(4)])