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