从数据框创建摘要

Question

我编写了 python 代码生成一个两列的数据框，其中包含我们员工的列表，以及他们当天分配到的团队（即 A 列：员工，B 列：团队).从这个 dataframe，我想生成一个 .txt 文件或格式如下的字符串：

团队 1（换行符） 工作人员（换行符） 工作人员（换行符） 工作人员（换行符） 第 2 队（换行符） 工作人员（换行符）

等等

我曾尝试使用 df.pivot() 来实现类似的效果，但运气不佳。抱歉，如果这是一个非常基本的问题，我正在学习所有这些。谢谢！

期望的输出：
获胜队伍
戴夫
亚历克斯
萨布丽娜
败北球队
约翰
道格拉斯

Answer 1

import pandas as pd

df = pd.DataFrame({"Staff":["Julie", "Max", "Sam", "Paula", "Mary"],
               "Team": ["Team 1", "Team 2", "Team 3", "Team 2", "Team 1"]})
grouped = df.groupby("Team")
result_string = ""
for team, group in grouped:
    result_string += (team + "\n")
    for person in group["Staff"]:
        result_string += ("\t" + person + "\n")

print(f'Dataframe: \n{df}')
print('______________\nResult String:')
print(result_string)

打印以下内容：

从数据框创建摘要

Creating a Summary from a Dataframe

python

pandas

txt