返回 function 輸出到 excel pandas

Question

我在這里有一個 function ，我想將輸出返回到一個 xlsx 文件，每個“流派”都有單獨的選項卡。 我試圖將每個“get_data”放入一個 df 然后連接所有 df，但這沒有用。 我有點卡在這里，不勝感激任何幫助/指導！

Answer 1

您只需創建一個pandas.ExcelWriter實例並使用DataFrame.to_excel即可將 pandas 數據框寫入 ZBF507C902E4D25

with pd.ExcelWriter('my-data.xlsx', engine='xlsxwriter') as writer:
    df1.to_excel(writer, sheet_name='Sheet1')
    df2.to_excel(writer, sheet_name='Sheet2')
    df3.to_excel(writer, sheet_name='Sheet3')

這將創建一個 excel 文件，並將在其每個工作表中寫入不同的數據幀。 請記住，您需要使用 pip 安裝xlsxwriter 。

pip install xlsxwriter

考慮到您提供的代碼，您需要進行一些調整才能實現目標。

• 更改get_data以返回數據。

   def get_data(pl_type_main, pl_type_sub, playlist_key): #... return pop_final

• 獲取每個特定類型的所有數據，將它們垂直堆疊在一起並構建另一個數據框。

   df1 = get_data('EDGE', 'POP') df2 = get_data('PULSE', 'POP') df3 = get_data('GENRE', 'POP') df_pop = pd.concat([df1, df2, df3], axis=0)

  對所有其他類型執行此操作，最后，您將擁有 4 個這樣的數據幀（即每個類型的數據幀）。

• 最后，將這些寫入 excel 文件，每個數據幀將是 excel 文件中的一張表。

   with pd.ExcelWriter('genres-data.xlsx', engine='xlsxwriter') as writer: df_pop.to_excel(writer, sheet_name='POP') df_rock.to_excel(writer, sheet_name='ROCK') # Similary all the others too