代码之家 › 专栏 › 技术社区 › SuperCiocia

同一条目的多个实例的numpy.searchsorted-python

search numpy python

SuperCiocia · 技术社区 · 7 年前

将numpy作为np导入

gens = np.array([2, 1, 2, 1, 0, 1, 2, 1, 2])
p = [0,1]

gens 匹配 p .

result = [[4],[2,3,5,7],[0,2,6,8]] 
#[[where matched 0], [where matched 1], [the rest]]

indx = gens.argsort()
res = np.searchsorted(gens[indx], [0])
gens[res] #gives 4, which is the position of 0

但我试着

indx = gens.argsort()
res = np.searchsorted(gens[indx], [1])
gens[res] #gives 1, which is the position of the first 1.

所以:

如何搜索出现多次的条目
倍数

1 回复 | 直到 7 年前

Andreas K. 7 年前

你可以用 np.where

>>> np.where(gens == p[0])[0]
array([4])

>>> np.where(gens == p[1])[0]
array([1, 3, 5, 7])

>>> np.where((gens != p[0]) & (gens != p[1]))[0]
array([0, 2, 6, 8])

或者 np.in1d 和 np.nonzero

>>> np.nonzero(np.in1d(gens, p[0]))[0]

>>> np.nonzero(np.in1d(gens, p[1]))[0]

>>> np.nonzero(~np.in1d(gens, p))[0]

推荐文章

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

1 年前

unfolx · numpy数组不等式的执行时间

1 年前

mchaudh4 · 用numpy表示三对角矩阵

1 年前

Geremia · 2D NumPy数组+1D数组?

1 年前

LMC · Numpy数组布尔索引以获取包含元素

1 年前

HJA24 · 根据条件用值正向填充Numpy矩阵/掩码

1 年前

alpelito7 · 如何只获取numpy数组中每个递增值的第一次出现?

1 年前

Amarth Gûl · 找到一组向量的最近收敛点

1 年前

Mr. W · numpy.divide是函数、类还是其他什么?

1 年前

Mr. W · 为什么numpy.array在编辑内部数据时如此缓慢?

1 年前