估計(jì)有不少小伙伴在將 DataFrame導(dǎo)入到Excel的時(shí)候，遇到過(guò)下面這種尷尬的情況：想給一個(gè)現(xiàn)有的Excel文件追加一個(gè)sheet，結(jié)果發(fā)現(xiàn)其它的sheet都沒(méi)了等，本文就來(lái)告訴你如何解決這些問(wèn)題

楔子

估計(jì)有不少小伙伴在將 DataFrame 導(dǎo)入到 Excel 的時(shí)候，都遇到過(guò)下面這種尷尬的情況：

想將多個(gè) DataFrame 導(dǎo)入到一個(gè) Excel 文件的多個(gè) sheet 中，但是卻發(fā)現(xiàn)生成的 Excel 文件里面只有最后一個(gè) sheet；
想給一個(gè)現(xiàn)有的 Excel 文件追加一個(gè) sheet，結(jié)果發(fā)現(xiàn)其它的 sheet 都沒(méi)了，只剩下新追加的 sheet；

那么下面就來(lái)看看如何解決這些問(wèn)題。

同時(shí)導(dǎo)入多個(gè) sheet

如果想導(dǎo)入多個(gè) sheet，那么肯定不能使用原來(lái) to_excel("文件名") 的方式，那樣只會(huì)保留最后一個(gè) sheet。我們應(yīng)該使用類(lèi) ExcelWriter 實(shí)現(xiàn)：

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]})

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

#?因此我們需要生成這個(gè)文件，此時(shí)順便將第一個(gè)?DataFrame?導(dǎo)進(jìn)去
df1.to_excel("test.xlsx",?index=False,?sheet_name="a")

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

#?接下來(lái)還是調(diào)用to_excel,?但是第一個(gè)參數(shù)不再是文件名,?而是上面的writer
#?將剩下的兩個(gè)DataFrame寫(xiě)進(jìn)去
df2.to_excel(writer,?index=False,?sheet_name="b")
df3.to_excel(writer,?index=False,?sheet_name="c")

#?保存并關(guān)閉writer,?寫(xiě)入磁盤(pán)
writer.save()
writer.close()

執(zhí)行代碼，然后打開(kāi)文件看一下。

此時(shí)我們看到結(jié)果是沒(méi)有問(wèn)題的，當(dāng)然向已存在的 Excel 文件追加 sheet 也是同理。

覆蓋一個(gè) sheet

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

下面問(wèn)題來(lái)了，我們上面的 Excel 文件有 "a"、"b"、"c" 三個(gè) sheet，假設(shè)我們想將 "b" 這個(gè) sheet 覆蓋掉，應(yīng)該怎么做呢？可能有人認(rèn)為，在追加的時(shí)候還指定 sheet_name="b" 不就行了，然鵝答案是不行的。

我們看到如果已有同名 sheet，那么不會(huì)覆蓋，還是創(chuàng)建一個(gè)新的 sheet，并自動(dòng)在結(jié)尾處加一個(gè) 1。如果我們?cè)诖嘶A(chǔ)上再寫(xiě)入 "b" 這個(gè) sheet 的話(huà)，那么又會(huì)多出一個(gè)名為 "b2" 的sheet。所以最好的辦法是，在導(dǎo)入之前先將 sheet 刪除。

import?pandas?as?pd

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

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

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

df?=?pd.DataFrame({"name":?["古明地覺(jué)",?"古明地戀"]})
#?我們將?b?這個(gè)?sheet?給刪除了
#?所以再導(dǎo)入?"b"?的時(shí)候就不會(huì)出現(xiàn)?"b3"?了
#?當(dāng)然?"b1"?和?"b2"?也順便被我們給刪掉了
df.to_excel(writer,?index=True,?sheet_name="b")

writer.save()
writer.close()

我們看到 "b1"、"b2" 兩個(gè) sheet 就沒(méi)了，當(dāng)然我們刪除的還有 "b" 這個(gè)sheet，只不過(guò)又重新創(chuàng)建了，當(dāng)然數(shù)據(jù)也是我們創(chuàng)建的新數(shù)據(jù)。

另外可能有人發(fā)現(xiàn)多個(gè) sheet 的順序不再是原來(lái)的 "a"、"b"、"c"，這是因?yàn)樵趧h除 "b" 之后，"a" 和 "c" 就靠在一起了，所以新寫(xiě)入 "b" 的時(shí)候就排在 "c" 的后面了，當(dāng)然個(gè)人覺(jué)得這沒(méi)有什么太大影響。

以上就是解決Pandas生成Excel時(shí)的sheet問(wèn)題的方法總結(jié)的詳細(xì)內(nèi)容，更多關(guān)于Pandas生成Excel sheet問(wèn)題的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: