python - 从未分配给变量的 pandas DataFrame 中选择行

Question

复杂的行选择表达式是通过创建一个布尔系列并将其传递给[]运算符来完成的，例如：

df_usa_2018 = df[(df.Year == 2018) & (df.Country == 'USA')]

但是，这需要将数据框分配给变量。是否有过滤数据框表达式而不分配它的快捷方式？我正在寻找的基本上是一个类似的功能

df_usa_2018 = pandas.read_csv(...).select(Year = 2018, Country = 'USA')

标签： pythonpandas

Answer 1

DataFrame.query与 change =to ==and&一起使用,：

df_usa_2018 = pandas.read_csv(...).query("Year == 2018 & Country == 'USA'")