numpy的argwhere,索引的使用错误

Question

刚学习玩numpy的argwhere的用法,根据字面意思返回的应当是符合条件的索引即

a = np.arange(6).reshape(2,3)
indix = np.argwhere(a>3)

#输出的应该是[1,1],[1,2]

问题是在已知索引的情况下如何根据索引来寻找到对应的元素

import numpy as np

a = np.arange(6).reshape(2,3)
indix = np.argwhere(a>3)
for i in indix:
    print(i)
    print(a[i])

结果报错，正常情况下应当如何根据索引寻找元素？

Answer 1

你好,这里a是rehshape成了(2,3)的形式，使用a[i]应该是不行的，可以调试看看indix的结构是什么，然后进行调整