python - 如何在单个过程中将（多个）分类特征转换为虚拟集

Question

我有一个包含很多分类特征（50+）的框架，我想将其转换为 0/1 存储桶（单热编码）。在python中，是否可以将它们全部转换而不必单独进行每一个？我尝试过 One-Hot-Encoder。它们都是字符串变量。

标签： pythonpandasmachine-learning

Answer 1

在pandas包中，您可以使用get_dummies（参考）：

import pandas as pd

pd.get_dummies(df[['string_column_a', 'string_column_b']])