Наткнулся на метод в pandas, который до сих пор мне был не известен, но очень пригодился, когда захотелось порефакторить свой старый ноутбук, чтобы повысить удобочитаемость. Это метод query. Он дает возможность выбрать строки датафрейма, исходя из определенного условия, но при этом не ссылаясь на сам датафрейм, что позволяет оформлять обработку данных в стиле method chaining.
К примеру, я хочу смерджить два датафрейма, а затем отфильтровать полученный результат по колонке из второго датафрейма.
Можно сделать это двумя отдельными командами:
df = df1.merge(df2,on='key',how='left')
df = df[df['column_from_df2'] == 1]
А вот как можно сделать это с помощью query:
df = df1.merge(df2,on='key',how='left').query("column_from_df2==1")
Также в query можно использовать переменные:
some_value = 1
df = df1.merge(df2,on='key',how='left').query("column_from_df2==@some_value")
Такие дела