python - with - seleccionar columnas de un dataframe pandas

Pandas DataFrame: aplica función a todas las columnas

Question

Puedo usar .map(func) en cualquier columna en un df, como:

df=DataFrame({''a'':[1,2,3,4,5,6],''b'':[2,3,4,5,6,7]})
df[''a'']=df[''a''].map(lambda x: x > 1)

Yo también podría:

df[''a''],df[''b'']=df[''a''].map(lambda x: x > 1),df[''b''].map(lambda x: x > 1)

¿Hay una forma más pitónica de aplicar una función a todas las columnas o al cuadro completo (sin un bucle)?

Answer 1

A partir de 0.20.0 adelante, puede usar la transform

In [578]: df.transform(lambda x: x > 1)
Out[578]:
A B C
0 False False False
1 False True False
2 False False True
3 False True True
4 False False False
In [579]: df
Out[579]:
A B C
0 -1 0 0
1 -4 3 -1
2 -1 0 2
3 0 3 2
4 1 -1 0

Y, para este caso simplista, ¿por qué no simplemente usar df > 1 ?

In [582]: df > 1
Out[582]:
A B C
0 False False False
1 False True False
2 False False True
3 False True True
4 False False False

Answer 2

Si entiendo bien, estás buscando el método de applymap .

>>> print df
A B C
0 -1 0 0
1 -4 3 -1
2 -1 0 2
3 0 3 2
4 1 -1 0
>>> print df.applymap(lambda x: x>1)
A B C
0 False False False
1 False True False
2 False False True
3 False True True
4 False False False