python - 根据现有列的分类值将列添加到数据框

Question

我有一个数据框，其中有一列包含变量名称，一列包含该变量的值。此表中的一些示例行可能是:

    id      variable    value   date
0   AS14.01 mood        6.000   2014-02-26
1   AS14.01 acti        0.091   2014-03-20
2   AS14.01 pers        0.101   2014-04-21

我想将其转换为一个数据框，其中变量列中的每个唯一值形成一个新列，其中包含相应行的值。对于特定 ID，每个日期应该只有 1 行。上面给出的数据框的示例行是:

    id      date        mood   acti   pers
0   AS14.01 2014-02-26  6.000  NaN    NaN
1   AS14.01 2014-03-20  NaN    0.091  NaN
2   AS14.01 2014-04-21  NaN    NaN    0.101

也许以前在stackoverflow上发过类似的问题，但我找不到这个问题的正确措辞。希望任何人都可以提供帮助。

Answer 1

这是使用 get_dummies 的一种方法。您也可以尝试使用旋转。这不会自动删除原始 variable 列。

df = df.join(pd.get_dummies(df.variable).replace(0, np.nan).mul(df.value, axis=0))


        # id variable  value        date   acti  mood   pers
# 0  AS14.01     mood  6.000  2014-02-26    NaN   6.0    NaN
# 1  AS14.01     acti  0.091  2014-03-20  0.091   NaN    NaN
# 2  AS14.01     pers  0.101  2014-04-21    NaN   NaN  0.101