【转】numpy-array自带的迭代器-----np.nditer

转自：https://www.jianshu.com/p/f2bd63766204

---

it = np.nditer(x, flags=['multi_index'], op_flags=['readwrite'])

查了查np.nditer原来是numpy array自带的迭代器。这里简单写个demo解释一下np.nditer的用法。

先构建一个3x4的矩阵

然后输入命令

flags=['multi_index']表示对a进行多重索引，具体解释看下面的代码。
op_flags=['readwrite']表示不仅可以对a进行read（读取），还可以write（写入），即相当于在创建这个迭代器的时候，我们就规定好了有哪些权限。

迭代一下试一试

 

print it.multi_index表示输出元素的索引，可以看到输出的结果都是index。
it.iternext()表示进入下一次迭代，如果不加这一句的话，输出的结果就一直都是(0, 0)。