Как в DataFrame сделать вычисляемое поле по условию?

Question

[Иван Мельников]

Например, вычисляемое поле
df['new'] = df.n / df.m
Проблема в том, что m может быть равно 0. Нужно, чтобы когда m = 0, то значение вычисляемого поля было равно 0.
Как это сделать?

Answer 1

[Alexeytur]

df['new'] = df.n / df.m if df.m != 0 else 0

Answer 2

[Михаил Кростелев]

Описывай функцию, которая принимает n и m и делит n на m. В случае если m==0 возвращает 0.

А потом просто вызывай эту функцию со своими аргументами:
df['new'] = foo(df.n, df.m)