Python Pandas Se il valore nella colonna B = uguale [X, Y, Z] sostituire colonna A con "T"

Question

Say Ho questa matrice:

A, B 
1, G 
2, X 
3, F 
4, Z 
5, I 

Se colonna B è uguale a [X, Y o Z] sostituire la colonna A con il valore "T"

Ho trovato come modificare i valori all'interno della stessa colonna ma non attraverso, qualsiasi aiuto sarebbe più apprezzato.

Answer 1

Si può provare questo:

import pandas as pd 
df = pd.DataFrame({ 
     'A': [1, 2, 3, 4, 5], 
     'B': ['G', 'X', 'F', 'Z', 'I'] 
    }) 
df.ix[df.B.isin(['X','Y','Z']), 'A'] = 'T' 
print df 

uscita:

A B 
0 1 G 
1 T X 
2 3 F 
3 T Z 
4 5 I 

Ricordate di usare ix o loc per evitare di impostare i valori su una fetta copiato.

Answer 2

Usa isin e loc per impostare il valore:

In [138]: 

df.loc[df.B.isin(['X','Y','Z']),'A']='T' 
df 
Out[138]: 
    A B 
0 1 G 
1 T X 
2 3 F 
3 T Z 
4 5 I 

È inoltre possibile utilizzare np.where:

In [140]: 

df['A'] = np.where(df.B.isin(['X','Y','Z']),'T', df['A']) 
df 
Out[140]: 
    A B 
0 1 G 
1 T X 
2 3 F 
3 T Z 
4 5 I