如何在数据类型为 int64 的 pandas 数据框列中进行以下替换?
-将 1 和 2 改为 0
-将 3 和 4 改为 1
-将 5、6、7 改为 2
所以最后我只有 3 个级别的分类?
答案 0 :(得分:1)
假设 df
是您的数据框的名称,而 column
是将您的列名作为字符串保存的变量的名称,那么:
fun = lambda data: 0 if data < 3 else 1 if data < 5 else 2
df[column] = df[column].apply(fun)
将转换列。