如何在数据帧中将多类标签更改为二进制标签

Question

我有一个标有类 0-3 的数据集，我正在尝试更改它，以便我可以使用二进制标签运行我的分类器。所以我想将 2 级和 3 级改为 1 级。有没有快速的方法？ 感谢您的帮助

Answer 1

我想你有一列包含 3 种类型，所以下面的代码只是简单地完成任务;

df['column'] = df['column'].map({'first_element':1, 'second_element':2,'third_element':3})

Answer 2

df['column'] = np.where(df.column == "class0", "class0", "class1")

这将有效地将 class0 保留为类 0，但其余类将被标记为 class1。