【转】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)

posted @ 2019-02-21 19:25  闲汉  阅读(2542)  评论(0编辑  收藏  举报