如何在熊猫中选择具有多个值的多行

Question

我有一个数据框和一个列表，如下所示。

         id     title     description
0  17810732  "nn nn."  "nnnn nnnn"
1  17810731  "mm mm."  "mmmm mmmm"
2  17810739  "ll ll."  "llll llll"
3  17810738  "jj jj."  "jjjj jjjj"

ids = [17810738, 17810731]

我想获取一个仅具有与ids列表相对应的行的数据框。

所以我的输出应该如下。

         id     title     description
0  17810738  "jj jj."  "jjjj jjjj"
1  17810731  "mm mm."  "mmmm mmmm"

我一直在使用此代码。

for id in ids:
   print(df.loc[df["id"] == id])

但是，它仅向每个id返回单独的数据帧，这不是我所需要的。

很高兴在需要时提供更多详细信息。

Answer 1

解决方案采用isin方法，所以

df[df['id'].isin(ids)]

可以解决问题。