numpy或者cupy里，有没有能批量返回一批数值的index的办法

水木社区手机版

主题:numpy或者cupy里，有没有能批量返回一批数值的index的办法
1楼|annals|2020-01-24 10:48:25|展开
就是np.where,如果参数Ndarray的dtype是bool,那么返回值的会是所有True的坐标

【在 ldy (俺是马甲) 的大作中提到: 】
: 采用循环，然后用np.where或者cp.where或者cp.argwhere的办法都只能一个一个的
: 这样如果要找一系列值的index的话就太慢了，有没有批量找的办法……
: 就像np.isin或者cp.isin这种就很好用啊
: ...................
--
FROM 221.219.140.89
3楼|annals|2020-01-24 11:48:18|展开
...你这是要从value找key吗? 感觉颠倒一下从key找value比较方便

【在 ldy (俺是马甲) 的大作中提到: 】
: 这个只适合一个个的，但是我想要的是通过矩阵操作出批量值
: 例如:
: import numpy as np
: ...................
--
FROM 221.219.140.89

BYR-Team©2010. KBS Dev-Team©2011 登录完整版