python - 将数据框列值转置为python中的列标题

Question

熊猫数据框具有三列（HHID、名称、模型）。我想将模型列值作为列标题。如果有与 HHID 列对应的模型值，“模型”中的值应该是“是”。

谁能帮我用python代码来做这个。

附上输入的截图。

输出应该是：

标签： pythonpython-3.xpandasdataframe

Answer 1

你可以试试pd.get_dummies

     hhid _NAME_    model
0   10006  flag4      NaN
1   10014  flag4      NaN
2   10047  flag4  make323
3   10055  flag4      NaN
4   10066  flag4  make268
5   10074  flag4      NaN
6   10085  flag4      NaN
7   10117  flag4  make146
8   10126  flag4      NaN
9   10136  flag4      NaN
10  10147  flag4  make148

>>> pd.get_dummies(df, columns=['model'])

     hhid _NAME_  model_make146  model_make148  model_make268  model_make323
0   10006  flag4              0              0              0              0
1   10014  flag4              0              0              0              0
2   10047  flag4              0              0              0              1
3   10055  flag4              0              0              0              0
4   10066  flag4              0              0              1              0
5   10074  flag4              0              0              0              0
6   10085  flag4              0              0              0              0
7   10117  flag4              1              0              0              0
8   10126  flag4              0              0              0              0
9   10136  flag4              0              0              0              0
10  10147  flag4              0              1              0              0

>>> # by setting prefix and prefix_sep to empty string
>>> # you could make your values as column names
>>> pd.get_dummies(df, columns=['model'], prefix='', prefix_sep='')

     hhid _NAME_  make146  make148  make268  make323
0   10006  flag4        0        0        0        0
1   10014  flag4        0        0        0        0
2   10047  flag4        0        0        0        1
3   10055  flag4        0        0        0        0
4   10066  flag4        0        0        1        0
5   10074  flag4        0        0        0        0
6   10085  flag4        0        0        0        0
7   10117  flag4        1        0        0        0
8   10126  flag4        0        0        0        0
9   10136  flag4        0        0        0        0
10  10147  flag4        0        1        0        0