如何使用 for 循環在循環的每次迭代中創建新的數據幀？

Question

目前，for 循環的每次迭代都被送入使用布爾索引的“扇區”數據框，並將結果導出到 excel。 “ia 部門”是一個包含字符串值的列表，例如醫療保健、consumer_staples、財產等。

for item in ia_sectors:
 sector = all_data[all_data["IA Sector"] == item]
 sector.to_excel(str(item)+".xlsx")

是否可以修改代碼，以便 for 循環的每次迭代都創建一個新的數據框，其中包含 item 中的迭代名稱及其來自布爾索引的各自迭代結果？ 例如

for item in ia_sectors:
 item0 = all_data[all_data["IA Sector"] == item0]
 sector.to_excel(str(item0)+".xlsx")
 item1 = all_data[all_data["IA Sector"] == item1]
 sector.to_excel(str(item1)+".xlsx")
 item2 = all_data[all_data["IA Sector"] == item2]
 sector.to_excel(str(item2)+".xlsx")

我想稍后在代碼中使用這些新數據框。

提前謝謝了。

Answer 1

國際大學聯盟，

dd = {}
ia_sector_data = all_data[all_date['IA Sector'].isin(ia_sectors)]

for n, g in ia_sector_data.groupby('IA Sector'):
    dd[n] = g
    g.to_excel(f'{n}.xls')

現在，您的數據框可以在字典 dd 中訪問。

使用 dd['Sec1'] 等等。