Вычислить разницу между строками в кадре данных Pandas, используя условную логику ⇐ Python

[Anonymous]

Я пытаюсь использовать функцию pandas.DataFrame.diff для вычисления разницы между строками в кадре данных. Загвоздка в том, что я хочу вычислить разницу только для определенных значений, используя некоторую простую условную логику, примененную к значению в столбце «данные» (т.е. если значение столбца «данные» равно 0, тогда установите разницу в 0. Обратите внимание, что это не так). должно быть установлено значение 0, его также можно установить как NaN).
Проблема аналогична вопросу «Как создать новый столбец, в котором значения выбираются на основе существующего столбца?». Однако мне не удалось применить это решение здесь.
Пример кода представлен ниже, а в таблице показаны текущие результаты и последний столбец с желаемым результатом.import pandas as pd

# Sample DataFrame
df = pd.DataFrame({'data': [10, 20, 0, 25, 0, 0, 30, 25, 40]})

df['diff'] = df.data.diff()

Таблица, показывающая результаты и желаемый результат в последнем столбце
value | diff | desired_result
10 NaN NaN
20 10 10
0 -20 0
25 25 25
0 -25 0
0 0 0
30 30 30
25 -5 -5
40 15 15

Я пытался получить следующий код с условной логикой, но это не работает
df['diff'] = df['data'].apply(lambda x: 0 if x == 0 else df.data.diff())


Подробнее здесь: https://stackoverflow.com/questions/792 ... onal-logic