Question

在以下Pandas DataFrame中，

df = pd.DataFrame({'session' : ["1","1","2","2","3","3"],
                'path'  : ["p1","p2","p1","p2","p2","p3"], 'seconds' : ["20","21","132","10","24","45"]})

我需要获得如下输出。（页面为列，会话为每个单元格中的行和秒。）

session,p1,p2,p3
1,20,21,0
2,132,10,0
3,0,24,45

到目前为止我做了什么。

In [76]: wordlist = ['p1', 'p2', 'p3']
In [77]: df2 = pd.DataFrame(df.groupby('session').apply(lambda x: ','.join(x.path)))
In [78]: df2 #I have renamed the columns
Out[78]: 

                  path
        session       
        1        p1,p2
        2        p1,p2
        3        p2,p3

In [79]: df3 = pd.DataFrame(df.groupby('session').apply(lambda x: ','.join(x.seconds.astype(str))))
In [80]: df3 #I have renamed the columns
Out[80]: 
                   path
        session        
        1         20,21
        2        132,10
        3         24,45

以下只给出了布尔结果。我需要得到我预期的输出。对此有何帮助？

In [84]: pd.DataFrame({name : df2["path"].str.contains(name) for name in wordlist})
Out[84]: 
            p1    p2     p3
session                    
1         True  True  False
2         True  True  False
3        False  True   True

Answer 1

使用数据透视表：

df.pivot(index='session', columns='path')

然后用零替换所有的Nan：

df2 = df1.fillna(0)

这为您提供以下输出：

        seconds        
path         p1  p2  p3
session                
1            20  21   0
2           132  10   0
3             0  24  45

然后你可能想要删除multiindex列：

df1.columns = df1.columns.droplevel(0)

产生你想要的解决方案（没有逗号）：

path      p1  p2  p3
session             
1         20  21   0
2        132  10   0
3          0  24  45

最后，您可以使用StringIO将其转换为以逗号分隔的字符串：

import StringIO
s = StringIO.StringIO()
df1.to_csv(s)
print s.getvalue()

使用以下输出：

session,p1,p2,p3
1,20,21,0
2,132,10,0
3,0,24,45

Pandas DataFrame格式化以获得预期输出

1 个答案: