python - Pandas 在数组中查找子数组的索引

Question

我有数组

a = numpy.array([1,3,5,7])
b = pandas.Series([1,2,3,4,5,6,7,8,9])

是否有快速命令来查找 b 中包含 a 中的值的所有匹配索引？例如

a in b = [0,2,4,6]

Answer 1

您可以使用 isin 来查找值在类数组对象中的位置(也适用于列表):

In [14]:

a = np.array([1,3,5,7])
b = pd.Series([1,2,3,4,5,6,7,8,9])
# call .index if you are just interested in the index values
b[b.isin(a)].index
Out[14]:
Int64Index([0, 2, 4, 6], dtype='int64')

如果不访问 .index 属性，您会得到一个返回的系列:

In [15]:

b[b.isin(a)]
Out[15]:
0    1
2    3
4    5
6    7
dtype: int64