我有一个数组，我需要获取满足条件为真和相同条件为假的索引，例如：
x = np.random.rand(100000000)
true_inds = np.where(x < 0.5)
false_inds = np.where(x >= 0.5)
在我的用例x中，该代码很大，在一个循环内被调用，而性能分析表明这np.where实际上是瓶颈。我目前正在执行类似于以上代码的操作，该代码不必要地扫描x两次以获得两组索引。是否有可能同时获得true_inds，并false_inds只用一个扫描x没有实施专门的替代np.where从头开始？
京公网安备 11010802040832号 | 京ICP备19059560号-6 