这是使用sklearn.preprocessing.LabelBinarizer的示例:
In [361]: from sklearn.preprocessing import LabelBinarizerIn [362]: lb = LabelBinarizer()In [363]: df['new'] = lb.fit_transform(df['ABC']).tolist()In [364]: dfOut[364]: Col1 ABC new0 XYZ A [1, 0, 0]1 XYZ B [0, 1, 0]2 XYZ C [0, 0, 1]
熊猫替代品:
In [370]: df['new'] = df['ABC'].str.get_dummies().values.tolist()In [371]: dfOut[371]: Col1 ABC new0 XYZ A [1, 0, 0]1 XYZ B [0, 1, 0]2 XYZ C [0, 0, 1]



