Разница в датафреймах Python с фильтром в одну строку

Question

Я пытаюсь найти разницу между двумя столбцами в кадре данных, но хочу применить фильтр.

Вот что я пытаюсь сделать

((df['column1']-df['column2'])/(df['totalcolumn'])).idxmax()

выше дает мне результат, но я хочу рассмотреть только строки, в которых Column1> 0 и Column2> 0.

Кто-нибудь может мне помочь?

Answer 1

df.query('column1 > 0 and column2 > 0') \
  .eval('(column1 - column2) / totalcolumn').idxmax()

Answer 2

Просто сначала получите подмножество интересующего вас кадра данных:

sub_df = df.loc[(df.column1 > 0) & (df.column2 > 0)]

(sub_df.column1 - sub_df.column2  / sub_df.totalcolumn).idxmax()

Если действительно важно, чтобы он был в одной строке (как подсказывает ваш заголовок), вы могли бы выполнить фильтрацию в той же строке, но это неэффективно и не элегантно ИМО:

(df.loc[(df.column1 > 0) & (df.column2 > 0), 'column1'] -
 df.loc[(df.column1 > 0) & (df.column2 > 0), 'column2'] /
 df.loc[(df.column1 > 0) & (df.column2 > 0), 'totalcolumn']).idxmax()