熊猫用计数创建新列

Question

我有一个类似的df：

将熊猫作为pd导入

df = pd.DataFrame({'car_id':[01,02,03,04,05],
               'crash_severinity' :[unknown, possible injury,not injury,possible injury, unknown]})

df

我希望它看起来像这样

df

df = pd.DataFrame({'car_id':[01,02,03,04,05],
               'unknown' :[1,0,0,0,1],
               'possible injury' :[0,1,0,1,0]
               'not injury:[0,0,1,0,0]})

先谢谢您！

Answer 1

这是您要寻找的：

df = pd.get_dummies(df)

输出为：

    crash_severity_not injured  crash_severity_possible injury  crash_severity_unknown
0                0                            0                           1
1                0                            1                           0
2                1                            0                           0
3                0                            1                           0
4                0                            0                           1