Python - 替换熊猫数据框列中的值

时间:2021-07-28 23:01:03

标签: python pandas dataframe

如何在数据类型为 int64 的 pandas 数据框列中进行以下替换?

-将 1 和 2 改为 0

-将 3 和 4 改为 1

-将 5、6、7 改为 2

所以最后我只有 3 个级别的分类?

1 个答案:

答案 0 :(得分:1)

假设 df 是您的数据框的名称,而 column 是将您的列名作为字符串保存的变量的名称,那么:

fun = lambda data: 0 if data < 3 else 1 if data < 5 else 2

df[column] = df[column].apply(fun)

将转换列。