從 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)

打印以下內容：