Python Pandas Nếu giá trị trong cột B = bằng [X, Y, Z] thay thế cột A với chữ "T"

Question

Nói rằng tôi có mảng này:

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

Nếu cột B tương đương với [X, Y hoặc Z] thay thế cột A bằng giá trị "T"

Tôi đã tìm thấy cách thay đổi giá trị trong cùng một cột nhưng không phải trên, bất kỳ trợ giúp nào sẽ được đánh giá cao nhất.

Answer 1

Bạn có thể thử này:

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 

Output:

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

Hãy nhớ sử dụng ix hoặc loc để tránh các giá trị thiết lập trên một lát sao chép.

Answer 2

Sử dụng isin và loc để thiết lập giá trị:

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 

Bạn cũng có thể sử dụng 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