将复杂列(类似字典)转换为多个列

时间:2018-02-13 13:34:59

标签: python pandas dictionary data-munging

我有一个包含两行和一列的数据框:

import pandas as pd
raw_data = {'T': [[(0, 0.5), (1, 0.25), (2, 0.25)], [(1, 0.99)]]}
df = pd.DataFrame(raw_data, columns=['T'])

#T 
#[(0, 0.5), (1, 0.25), (2, 0.25)]
#[(1, 0.99)]

如你所见,row是一种字典,我知道字典元素的最大数量(在我们的例子中,如果从0开始计数则为3或2))。 我想创建额外的三列 T0,T1,T2 ,并在此数据框中包含以下值:

T0   T1    T2 (df header)

0.5  0.25  0.25

0    0.99  0
  • 可能超过2行
  • 可能会有更多列,但不应影响问题

1 个答案:

答案 0 :(得分:2)

list comprehensionconcat一起使用,然后转置,fillnaadd_prefix

df=pd.concat([pd.Series(dict(x)) for x in raw_data['T']], axis=1).T.fillna(0).add_prefix('T')
print (df)
    T0    T1    T2
0  0.5  0.25  0.25
1  0.0  0.99  0.00