多个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)])
我们可以计算 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)])