假设您有一个ndarray z，并希望得到以下索引
z_idxes1 = np.argwhere(z == i)
z_idxes2 = np.argwhere(z != i)
我们可以在一次调用一些numpy函数而不是两次调用argwhere时执行此操作吗？
答案 0 :(得分：1)
不在一个调用中，但您可以使用逻辑非运算符来取消掩码数组。
mask = (z == i)
z_idxes1 = np.argwhere(mask)
z_idxes2 = np.argwhere(~mask)