У мене є DataFrame df
:
A B
a 2 2
b 3 1
c 1 3
Я хочу створити новий стовпець на основі таких критеріїв:
якщо рядок A == B: 0
якщо рядокA > B: 1
якщо рядок A < B: -1
так, враховуючи наведену вище таблицю, вона повинна бути:
A B C
a 2 2 0
b 3 1 1
c 1 3 -1
Для типових if else
випадків, які я роблю np.where(df.A > df.B, 1, -1)
, чи забезпечують панди спеціальний синтаксис для вирішення моєї проблеми одним кроком (без необхідності створювати 3 нові стовпці, а потім комбінувати результат)?
apply
і встановити, щоaxis=1
буде працювати, не впевнений, що я можу придумати операцію, яка дасть вам те, що ви хочете