熊猫无循环行索引

Question

我正在尝试使用pandas索引对行进行索引，但是似乎没有合适的方法为此输入列表。这是我尝试使用无循环的解决方案。

render_plots

有了循环，我就能得到想要的结果。

import pandas as pd


df = pd.DataFrame({'a': [1, 2, 3, 4, 9], 'b': [3, 3, 4, 5, 100]})

print(df)

interest = [3, 4]

# results = df['a'].eq(interest)
# results = df[(df['a'] == 3) & (df['a'] == 4)]

df(results)

# print(df[df['b'] == 3])  # index 0 1 2

我已经进行了搜索，但是大多数文档都将对列“ a”和“ b”进行索引和/或一次仅使用一个值进行索引，而不是列表。让我知道我是否不清楚

Answer 1

您想要.isin的IIUC吗？

>>> df[df.a.isin([3,4])]

---

   a  b
2  3  4
3  4  5