时间戳

首页 > 解决方案 > 多个numpy数组的最大值

python - 多个numpy数组的最大值

问题描述

我们可以计算 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：参数数量无效

毕竟，LNumpy 数组对象是一个 Python 列表，所以*L应该为np.maximum函数调用解包它。为什么它不起作用？

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

标签： pythonarraysnumpyunpackargument-unpacking

解决方案

正如评论中指出的，reduce这里的解决方案是：

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

推荐阅读