如何在Python2.7中基于条件转置特定列

Question

我在一个文件中有以下数据格式：

ID,var_name,var_value
1,ABC,This is abc1
1,DEF,This is def1
2,ABC,This is abc2
2,DEF,This is def2
2,GHI,This is ghi2
3,ABC,This is abc3
4,ABC,This is abc4
4,DEF,This is def4

我还有一个标题列表=['ABC','GHI']

在上面的数据集中，每个“ID”不一定包含所有的变量，但是ID:2包含了最大数量的变量（ABC、DEF、GHI）。我需要将上述数据集转换为以下嵌套列表格式：

^{pr2}$

这意味着清单应：

• 填充所有ID的数据
• 为上述数据集中不存在的变量创建空字符串。在
• 填充嵌套列表，保持与标题列表相同的顺序
• 只填充那些标题值，即标题列表只包含值“ABC”和“GHI”，因此嵌套列表只应填充值“ABC”和“GHI”，并应忽略上述数据集中的“DEF”行。在

我想在python2.7中使用Pandas。在

Answer 1

或者，您可以只设置multiindex和unstack：

In []:
L = ['ABC', 'GHI']
df = df.set_index(['ID', 'var_name'])['var_value'].unstack(fill_value='')[L].reset_index()
df

Out[]:
var_name  ID           ABC           GHI
0          1  This is abc1              
1          2  This is abc2  This is ghi2
2          3  This is abc3              
3          4  This is abc4              

In []:
[df.columns.tolist()] + df.values.tolist()

Out[]:
[['ID', 'ABC', 'GHI'],
 [1, 'This is abc1', ''],
 [2, 'This is abc2', 'This is ghi2'],
 [3, 'This is abc3', ''],
 [4, 'This is abc4', '']]