从 Dataframe 创建摘要

Question

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

团队 1 （换行）职员（换行）职员（换行）职员（换行）团队 2 （换行）职员（换行）

等等

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

所需的 Output：
获胜团队
戴夫
亚历克斯
萨布丽娜
失败的球队
约翰
道格拉斯

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)

打印以下内容：