在 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(使用前将#替换为@)