Как фильтровать значения после группового в пандах

Question

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

df.sort_values(by='col1').groupby('col2').apply(lambda x: x if x!=0).apply(list)

Однако, если говорит invalid syntax.Что я делаю не так?

Answer 1

Превратите первую функцию в apply в lambda и выполните там фильтрацию:

(df.sort_values(by='col1')
   .groupby('col2')
   .apply(lambda x: [y for y in x if y != 0]))