Eu tenho um DataFrame df
:
A B
a 2 2
b 3 1
c 1 3
Quero criar uma nova coluna com base nos seguintes critérios:
se linha A == B: 0
se linhaA > B: 1
se linha A < B: -1
então, dada a tabela acima, deveria ser:
A B C
a 2 2 0
b 3 1 1
c 1 3 -1
Para if else
casos típicos que eu faço np.where(df.A > df.B, 1, -1)
, o pandas fornece uma sintaxe especial para resolver meu problema com uma etapa (sem a necessidade de criar 3 novas colunas e, em seguida, combinar o resultado)?
apply
e definiraxis=1
funcionaria, não tenho certeza se posso pensar em uma operação que daria a você o que você deseja