重命名熊猫中的列值

Question

我有一个名为region的数据集。样本值是 例如。 region_1, region_2, region_3等

我需要将这些值替换为 例如。 1,2,3等

任何特定功能可以处理这种简单的转换吗？

谢谢

Answer 1

我认为您需要使用选择的第二个值进行分割，并在必要时转换为整数：

df.region = df.region.str.split('_').str[1].astype(int)

或将extract与正则表达式一起使用以提取整数：

df.region = df.region.str.extract('(\d+)', expand=False).astype(int)

示例：

df = pd.DataFrame({'region':['region_1','region_2','region_3']})
df.region = df.region.str.extract('(\d+)', expand=False).astype(int)
print (df)
   region
0       1
1       2
2       3