如何通過檢查條件來替換dataFrame中的值？

Question

我有以下數據框：

A  |   Date1  | Date2 
10 | 2/2/2016 | 3/2/2016
11 | 1/5/2016 | 1/5/2016
12 | 2/3/2016 | 2/3/2016
13 | 1/5/2016 | 3/2/2013

如果Date1中的值等於Date2 ，我想將A列中的值設為0 。 最終結果：

A  |   Date1  | Date2 
10 | 2/2/2016 | 3/1/2016
0  | 1/5/2016 | 1/5/2016
0  | 2/3/2016 | 2/3/2016
13 | 1/5/2016 | 3/2/2013

我想這樣做而不編寫for循環。 我可以使用申請嗎？

您可以重新創建我的df：

df = pd.DataFrame([[10, "2/2/2016", "3/2/2016" ] , [11, "1/5/2016", "1/5/2016"] , [12 , "2/3/2016" , "2/3/2016" ] , [13, "1/5/2016", "3/2/2013"]])
df.columns = ['A','B','C']

Answer 1

使用mask ：

import pandas as pd


df = pd.DataFrame([[10, "2/2/2016", "3/2/2016" ] ,
                   [11, "1/5/2016", "1/5/2016"] ,
                   [12 , "2/3/2016" , "2/3/2016" ] ,
                   [13, "1/5/2016", "3/2/2013"]])
df.columns = ['A','B','C']
print (df)
    A         B         C
0  10  2/2/2016  3/2/2016
1  11  1/5/2016  1/5/2016
2  12  2/3/2016  2/3/2016
3  13  1/5/2016  3/2/2013

df['A'] = df.mask(df.B == df.C, 0)
print (df)
    A         B         C
0  10  2/2/2016  3/2/2016
1   0  1/5/2016  1/5/2016
2   0  2/3/2016  2/3/2016
3  13  1/5/2016  3/2/2013

解where ：

df['A'] = df.where(df.B != df.C, 0)
print (df)
    A         B         C
0  10  2/2/2016  3/2/2016
1   0  1/5/2016  1/5/2016
2   0  2/3/2016  2/3/2016
3  13  1/5/2016  3/2/2013

Answer 2

使用jezrael的設置

設定

import pandas as pd


df = pd.DataFrame([[10, "2/2/2016", "3/2/2016" ] ,
                   [11, "1/5/2016", "1/5/2016"] ,
                   [12 , "2/3/2016" , "2/3/2016" ] ,
                   [13, "1/5/2016", "3/2/2013"]])
df.columns = ['A','B','C']

使用loc

df.loc[df.B == df.C, 'A'] = 0

print df

    A         B         C
0  10  2/2/2016  3/2/2016
1   0  1/5/2016  1/5/2016
2   0  2/3/2016  2/3/2016
3  13  1/5/2016  3/2/2013