¿Cómo debo tomar el máximo de 2 columnas en un marco de datos y convertirlo en otra columna?

Question

Tengo un marco de datos con las columnas A, B. Necesito crear una columna C tal que para cada registro/fila:

C = max(A, B).

¿Cómo debo hacer esto?

Gracias.

Answer 1

Usted puede obtener el máximo de esta manera:

>>> import pandas as pd 
>>> df = pd.DataFrame({"A": [1,2,3], "B": [-2, 8, 1]}) 
>>> df 
    A B 
0 1 -2 
1 2 8 
2 3 1 
>>> df[["A", "B"]] 
    A B 
0 1 -2 
1 2 8 
2 3 1 
>>> df[["A", "B"]].max(axis=1) 
0 1 
1 8 
2 3 

y así:

>>> df["C"] = df[["A", "B"]].max(axis=1) 
>>> df 
    A B C 
0 1 -2 1 
1 2 8 8 
2 3 1 3 

Si sabe que "A" y "B" son las únicas columnas, incluso se podría conseguir lejos con

>>> df["C"] = df.max(axis=1) 

Y se podría utilizar .apply(max, axis=1) también, supongo.