获取 numpy 数组的多个轴的平均值

2024-05-12

在 numpy 中是否有一种快速计算多轴平均值的方法?我正在计算 n 维数组除 0 轴之外的所有轴的平均值。

我目前正在做这个;

for i in range(d.ndim - 1):
    d = d.mean(axis=1)

我想知道是否有一个不使用 python 循环的解决方案。


在 numpy 1.7 中你可以给多个轴np.mean:

d.mean(axis=tuple(range(1, d.ndim)))

我猜测这将与其他建议的解决方案类似,除非重塑数组以展平所有维度会触发数据副本,在这种情况下,这应该要快得多。所以这可能会带来更一致的性能。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

获取 numpy 数组的多个轴的平均值 的相关文章

随机推荐