自定义nditer行为

Question

我正在使用一些形状（x，y，3）的图像数据。是一种使用像nditer这样的numpy结构迭代对应于像素的（r，g，b）元组的方法吗？开箱即用，nditer迭代标量值，但许多numpy函数有类似axis=参数的东西来改变这样的行为。

Answer 1

检查np.ndindex。它的文档可能足以让你入门。更好的是，阅读它的代码。

我基本上通过反向工程ndindex回答了类似的问题。如果Related侧边栏中没有链接，我会快速搜索。

shallow iteration with nditer