日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

解決Pandas生成Excel時的sheet問題的方法總結_python

作者:古明地覺 ? 更新時間: 2022-10-04 編程語言

楔子

估計有不少小伙伴在將 DataFrame 導入到 Excel 的時候,都遇到過下面這種尷尬的情況:

  • 想將多個 DataFrame 導入到一個 Excel 文件的多個 sheet 中,但是卻發現生成的 Excel 文件里面只有最后一個 sheet;
  • 想給一個現有的 Excel 文件追加一個 sheet,結果發現其它的 sheet 都沒了,只剩下新追加的 sheet;

那么下面就來看看如何解決這些問題。

同時導入多個 sheet

如果想導入多個 sheet,那么肯定不能使用原來?to_excel("文件名")?的方式,那樣只會保留最后一個 sheet。我們應該使用類 ExcelWriter 實現:

import?pandas?as?pd

df1?=?pd.DataFrame({"a":?[1,?2],?"b":?[3,?4]})
df2?=?pd.DataFrame({"a":?[2,?3],?"b":?[4,?5]})
df3?=?pd.DataFrame({"a":?[3,?4],?"b":?[5,?6]})

#?調用pd.ExcelWriter,?需要指定mode="a",?engine="openpyxl"
#?注意:?將mode設置為"a"表示追加,?但是它要求文件必須存在,?否則報錯
"""
writer?=?pd.ExcelWriter("test.xlsx",?mode="a",?engine="openpyxl")
"""

#?因此我們需要生成這個文件,此時順便將第一個?DataFrame?導進去
df1.to_excel("test.xlsx",?index=False,?sheet_name="a")

#?然后再實例化ExcelWriter
writer?=?pd.ExcelWriter("test.xlsx",?mode="a",?engine="openpyxl")

#?接下來還是調用to_excel,?但是第一個參數不再是文件名,?而是上面的writer
#?將剩下的兩個DataFrame寫進去
df2.to_excel(writer,?index=False,?sheet_name="b")
df3.to_excel(writer,?index=False,?sheet_name="c")

#?保存并關閉writer,?寫入磁盤
writer.save()
writer.close()

執行代碼,然后打開文件看一下。

此時我們看到結果是沒有問題的,當然向已存在的 Excel 文件追加 sheet 也是同理。

覆蓋一個 sheet

向 Excel 文件同時寫入多個sheet,以及追加sheet,我們已經知道該怎么做了,然后是覆蓋 sheet。首先我們覆蓋 sheet 的時候還要保證其它 sheet 不受影響,所以 mode 仍然要設置為追加模式。

下面問題來了,我們上面的 Excel 文件有 "a"、"b"、"c" 三個 sheet,假設我們想將 "b" 這個 sheet 覆蓋掉,應該怎么做呢?可能有人認為,在追加的時候還指定 sheet_name="b" 不就行了,然鵝答案是不行的。

我們看到如果已有同名 sheet,那么不會覆蓋,還是創建一個新的 sheet,并自動在結尾處加一個 1。如果我們在此基礎上再寫入 "b" 這個 sheet 的話,那么又會多出一個名為 "b2" 的sheet。所以最好的辦法是,在導入之前先將 sheet 刪除。

import?pandas?as?pd

writer?=?pd.ExcelWriter("test.xlsx",?mode="a",?
????????????????????????engine="openpyxl")
wb?=?writer.book
#?pandas操作Excel底層也是依賴于其它的模塊,?比如xlrd、openpyxl
#?所以這里的?wb?=?writer.book??就相當于
"""
from?openpyxl?import?load_workbook
wb?=?load_workbook("test.xlsx")
"""

#?查看已存在的所有的sheet,?總共是5個
#?其中?"b1"和"b2"?是自動創建的
print(wb.sheetnames)??#?['a',?'b',?'c',?'b1',?'b2']

#?下面我們來刪除sheet
wb.remove(wb["b1"])
wb.remove(wb["b2"])
wb.remove(wb["b"])

df?=?pd.DataFrame({"name":?["古明地覺",?"古明地戀"]})
#?我們將?b?這個?sheet?給刪除了
#?所以再導入?"b"?的時候就不會出現?"b3"?了
#?當然?"b1"?和?"b2"?也順便被我們給刪掉了
df.to_excel(writer,?index=True,?sheet_name="b")

writer.save()
writer.close()

我們看到 "b1"、"b2" 兩個 sheet 就沒了,當然我們刪除的還有 "b" 這個sheet,只不過又重新創建了,當然數據也是我們創建的新數據。

另外可能有人發現多個 sheet 的順序不再是原來的 "a"、"b"、"c",這是因為在刪除 "b" 之后,"a" 和 "c" 就靠在一起了,所以新寫入 "b" 的時候就排在 "c" 的后面了,當然個人覺得這沒有什么太大影響。

原文鏈接:https://mp.weixin.qq.com/s/Omo2e0PMAo_rtF7GSWGCFw

欄目分類
最近更新