Создание нового столбца на основе оператора if другого столбца, извлекающего данные из двух разных столбцов

Question

У меня есть данные, которые выглядят так:

A    B    C    D
3    1    4    0.5
4    2    6    0.25
2    1    3   -0.4
3    3    6   -0.8

И я хотел бы добавить столбец E, который является результатом либо A, разделенного на C, либо B, разделенного на C на основе если D <0. Я подумал, что должен использовать apply в этой ситуации, но я не знаю, как работать как с al oop, так и с оператором if в apply (). Любые предложения о том, как справиться с этим? </p>

Answer 1

np.where может помочь здесь: Pandas условное создание столбца серии / фрейма данных

df['E'] = np.where(df['D'] < 0, df['A']/df['C'], df['B']/df['C'])