2012-10-05 43 views
23

que puedo usar .map(func) en cualquier columna en un df, como:pandas trama de datos: aplicar la función a todas las columnas

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) 

pude también:

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

¿Existe una manera más Pythonic aplicar una función para todas las columnas o el cuadro completo (sin un bucle)?

+0

simplificar su '' lambda' a lambda x: x> 1' – Blender

+0

@ Blender - gracias, editado ... – root

+0

Sólo señalarlo. Realmente no necesita editar la pregunta original. – Blender

Respuesta

35

Si entiendo bien, está buscando el método 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 
+0

@ BrenBarn - sí, esto es exactamente lo que estaba buscando. no lo noté de los documentos. Gracias. – root

1

De 0.20.0 en adelante, puede utilizar 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 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 
Cuestiones relacionadas