python - pandas 中类似 SQL 的语句？

Question

这个问题已经有答案了:

index a Python Pandas dataframe with multiple conditions SQL like where statement (1 个回答)

已关闭 7 年前。

我试图在 pandas 中运行以下 SQL 语句(显然是在 Python 代码中)，但一无所获:

select year, contraction, indiv_count, total_words from dataframe
    where contraction in ("i'm","we're","we've","it's","they're")

哪里contractions是 char ，和year , indiv_count ，和total_words是int .

我对 Pandas 不太熟悉。如何在 Python 中创建类似的语句？

Answer 1

如果您还没有阅读 Anton 评论中列出的文档，我建议您阅读，但它缺少 .isin() 方法的文档，这是您复制 SQL 所需的中。

df[df['contraction'].isin(["i'm","we're","we've","it's","they're"])]

然后可以使用 .loc[] 或任何您最喜欢的方法(有很多)来获取列选择。