熊猫中的类似 SQL 的语句？ [复制]

Question

我正在尝试在 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。

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

Answer 1

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

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

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