【发布时间】:2020-12-01 17:47:49
【问题描述】:
我有一个由一些早期代码生成的 numpy 数组,需要逐块添加。
例如,数组 a 有 32 个条目,应该添加新数组 b 有 4 个条目,b[0] 有数字 0-7,b[1] 有 8-15,依此类推。请参阅下面的示例,了解如何手动操作。
import numpy as np
a = np.random.rand(32)
b = np.zeros(4)
b[0] = np.sum(a[0:8])
b[1] = np.sum(a[8:16])
b[2] = np.sum(a[16:24])
b[3] = np.sum(a[24:32])
现在我知道我可以使用 for 循环进行求和,但我希望有更花哨的东西,因为我正在处理相当大的数组。我知道 numpy 强大的切片魔法,所以我想可以使用这样的东西。
【问题讨论】: