Python - 替換熊貓數據框列中的值

Question

如何在數據類型為 int64 的 Pandas 數據框列中進行以下替換？

- 將 1 和 2 更改為 0

- 將 3 和 4 改為 1

- 將 5、6、7 改為 2

所以最后我只有 3 個級別進行分類？

Answer 1

假設df是您的數據框的名稱，而column是將您的列名作為字符串保存的變量的名稱，那么：

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

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

將轉換列。