在大小为n*n python的ndarray中接收到大于x元素的所有索引

Question

我有一个大小为n*n的ndarray（200200）。在

我想找到所有大于0的值的索引，然后将它们更改为1。在

我试着做那样的事，但我得到了例外。在

toggleOn_index = [x for x in  net if x>0]
net[toggleOn_index]=1

最有效的方法是什么？在

Answer 1

如果我从数组a开始，如下所示

>>> a = np.random.rand(5, 5) - 0.5
>>> a
array([[ 0.24116601, -0.23917393,  0.46607471,  0.08560729,  0.33878893],
       [ 0.24681769, -0.01693847,  0.22579766, -0.38570793, -0.2075437 ],
       [-0.19602781, -0.46108244, -0.40341244,  0.42579818,  0.4655471 ],
       [-0.14669409,  0.36171014, -0.36496558, -0.30983999, -0.06633584],
       [-0.09159042,  0.23346471,  0.37680536, -0.09281205, -0.04552193]])

我可以使用

^{pr2}$