如何在用户定义的块中迭代Numpy数组？

Question

我有一个numpy数组，我想迭代它。现在，我知道了nditer（）函数，它在数组上循环，每次取一个点。但是，我希望一次遍历数组，获取'n'个点。这里n的范围在100-200之间。有没有关于如何做到这一点的参考资料

Answer 1

一个简单的解决方案是重塑

for chunk in arr.reshape(-1,n):
   ...