python - 使用列中的数据创建新列

Question

这是我在 pandas 中的数据

      Movie        Tags
0  War film  tank;plane
1  Spy film   car;plane

我想用带有 0 和 1 的标签列创建新列，并在列的名称中添加前缀，如“T_”。

喜欢:

      Movie        Tags T_tank T_plane T_car
0  War film  tank;plane      1       1     0
1  Spy film   car;plane      0       1     1

例如，我有一些关于如何使用 split(";") 和 df.loc[:,'T_plane'] 逐行执行此操作的想法。但我认为这可能不是最佳方式。

问候

Answer 1

使用 sklearn 库:

from sklearn.preprocessing import MultiLabelBinarizer

mlb = MultiLabelBinarizer()

res = df.join(pd.DataFrame(mlb.fit_transform(df['Tags'].str.split(';')),
                           columns=mlb.classes_).add_prefix('T_'))

print(res)

      Movie        Tags  T_car  T_plane  T_tank
0  War film  tank;plane      0        1       1
1  Spy film   car;plane      1        1       0