Python Pandas替換字符串

Question

給定此數據框...：

DF = pd.DataFrame({'COL1': ['A', 'B', 'C', 'D','D','D'], 
                   'COL2': [11032, 1960, 11400, 11355, 8, 7], 
                   'year': ['2016', '2017', '2018', '2019', '2020', '2021']})
DF

   COL1 COL2    year
0   A   11032   2016
1   B   1960    2017
2   C   11400   2018
3   D   11355   2019
4   D   8       2020
5   D   7       2021

如果滿足以下條件：

1. COL1 ='D'
2. year ='2021'（請視為字符串）

然后將COL2的值更改為100 。

另外，如果將year當作int怎么辦？

Answer 1

這是loc的典型用例。 為行選擇選擇多個條件時， cond 1 and (cond 2 or cond 3)模式的示例為df.loc[(condition 1) & ((condition 2) | (condition 3), 'selected columns'] 。

DF.loc[(DF.COL1 == 'D') & (DF.year == '2021'), 'COL2'] = 100

如果年份是整數，則可以將其修改如下，以同時捕獲整數和字符串：

DF.loc[(DF.COL1 == 'D') & (DF.year.astype(int) == 2021), 'COL2'] = 100