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

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 留给 class0，而其余的类将被标记为 class1。