Python - Pandas - 根据分类值将多列的行组合成数据框中的单行

Question

我正在处理 Python 3.4 中涉及 Pandas 的问题。我被困在一个涉及重新组织我的数据框的小部分。我会更具体。

我有一个名为“模型”的表格，格式为：

Model Input

我希望得到所需的输出形式相当于：

我希望得到类似的输出：

Desired Output

我已经调查过Convert a python dataframe with multiple rows into one row using python pandas? 和How to combine multiple rows into a single row with pandas。我对是否应该使用 groupby 或数据透视表感到困惑。我尝试使用两者，但要么得到 KeyError，要么得到我想要的正确格式。是否有任何特定的库可以帮助完成上述任务？

Answer 1

您可以使用 groupby 并申请：

num_V = 5
max_row = df.groupby('ID').ID.count().max()
df2= (
        df.groupby('ID')
        .apply(lambda x: x.values[:,1:].reshape(1,-1)[0])
        .apply(pd.Series)
        .fillna(0)
)

df2.columns = ['V{}_{}_{}'.format(i+1,j,i) for j in range(max_row) for i in range(num_V)]