我有一个包含两行和一列的数据框:
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
答案 0 :(得分:2)
将list comprehension
与concat
一起使用,然后转置,fillna
和add_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