以 csv 格式显示 Pandas DataFrame

Question

我有一个熊猫数据框q2，看起来像这样：

    StudentID     Subjects
6         323      History
9         323      Physics
8         999    Chemistry
7         999      History
4         999      Physics
0        1234    Chemistry
5        2834      Physics
1        3455    Chemistry
2        3455      History
10       3455  Mathematics
3       56767  Mathematics

我想知道哪个学生上过哪些课程并显示在屏幕上。

gb = q2.groupby(('StudentID'))
result = gb['Subjects'].unique()

c1=pd.DataFrame({'StudentID':result.index, 'Subjects':result.values})

c1 看起来像这样

   StudentID                           Subjects
0        323                 [History, Physics]
1        999      [Chemistry, History, Physics]
2       1234                        [Chemistry]
3       2834                          [Physics]
4       3455  [Chemistry, History, Mathematics]
5      56767                      [Mathematics]

但是，所需的输出如下：

323: History, Physics
999: Chemistry, History, Physics
1234: Chemistry
2834: Physics
3455: Chemistry, History, Mathematics
56767: Mathematics

我能做什么？

Answer 1

我认为你可以 apply 函数 join。也可以使用reset_index 创建DataFrame：

gb = q2.groupby(('StudentID'))
result = gb['Subjects'].unique()

c1 = result.reset_index()

c1.Subjects = c1.Subjects.apply(', '.join)
print (c1)
   StudentID                         Subjects
0        323                 History, Physics
1        999      Chemistry, History, Physics
2       1234                        Chemistry
3       2834                          Physics
4       3455  Chemistry, History, Mathematics
5      56767                      Mathematics

最后你可以将列StudentID 转换为str（如果dtype 是int）并连接在一起：

c1['new'] = c1.StudentID.astype(str) + ':' + c1.Subjects
print (c1)
   StudentID                         Subjects  \
0        323                 History, Physics   
1        999      Chemistry, History, Physics   
2       1234                        Chemistry   
3       2834                          Physics   
4       3455  Chemistry, History, Mathematics   
5      56767                      Mathematics   

                                    new  
0                  323:History, Physics  
1       999:Chemistry, History, Physics  
2                        1234:Chemistry  
3                          2834:Physics  
4  3455:Chemistry, History, Mathematics  
5                     56767:Mathematics  

如果可以覆盖原始数据，请使用：

result = result.index.to_series().astype(str) + ':' + result.apply(', '.join)
print (result)
StudentID
323                      323:History, Physics
999           999:Chemistry, History, Physics
1234                           1234:Chemistry
2834                             2834:Physics
3455     3455:Chemistry, History, Mathematics
56767                       56767:Mathematics
dtype: object