gpt4 book ai didi

python - 如何通过numpy同时获取两种类型的索引?

转载 作者:行者123 更新时间:2023-12-01 09:28:08 26 4
gpt4 key购买 nike

假设您有一个 ndarray z,并且希望获得以下索引

z_idxes1 = np.argwhere(z == i)
z_idxes2 = np.argwhere(z != i)

我们可以通过一次调用某个 numpy 函数而不是两次调用 argwhere 来完成此操作吗?

最佳答案

不是在一次调用中,但您可以使用逻辑非运算符来获取掩码数组的否定。

mask = (z == i)
z_idxes1 = np.argwhere(mask)

z_idxes2 = np.argwhere(~mask)

关于python - 如何通过numpy同时获取两种类型的索引?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50199997/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com