我有一个numpy的一维数组,我想找到一个值超过numpy数组中的值的索引的位置.
例如
aa = range(-10,10)
Run Code Online (Sandbox Code Playgroud)
找到位置,超出aa价值的位置5.
我需要弄清楚如何在2d numpy数组中找到值的所有索引.
例如,我有以下2d数组:
([[1 1 0 0],
[0 0 1 1],
[0 0 0 0]])
Run Code Online (Sandbox Code Playgroud)
我需要找到所有1和0的索引.
1: [(0, 0), (0, 1), (1, 2), (1, 3)]
0: [(0, 2), (0, 3), (1, 0), (1, 1), (the entire all row)]
Run Code Online (Sandbox Code Playgroud)
我试过这个,但它没有给我所有的索引:
t = [(index, row.index(1)) for index, row in enumerate(x) if 1 in row]
Run Code Online (Sandbox Code Playgroud)
基本上,它只给我每行中的一个索引[(0, 0), (1, 2)].