作者：黃偉
來源：杰哥的IT之旅（ID：Jake_Internet）
轉載請聯(lián)系授權（WeChat ID：Hc220066）

一、相關知識點講解

1、需要使用的相關庫

1import?numpy?as?np
2import?pandas?as?pd
3import?os

2、os.walk(pwd)

舉例如下
① 先來看看"G:\a"文件夾下有哪些東西。

② 代碼操作如下：

1pwd?=?"G:\\a"
2print(os.walk(pwd))
3for?i?in?os.walk(pwd):
4????print(i)
5for?path,dirs,files?in?os.walk(pwd):
6????print(files)

③ 結果如下：

1object?walk?at?0x0000029BB5AEAB88>
2('G:\\a',?[],?['aa.txt',?'bb.xlsx',?'cc.txt',?'dd.docx'])
3['aa.txt',?'bb.xlsx',?'cc.txt',?'dd.docx']

3、os.path.join(path1,path2…)

作用：將多個路徑組合后返回

舉例如下

1path1?=?'G:\\a'
2path2?=?'aa.txt'
3print(os.path.join(path1,path2))

結果如下：

1G:\a\aa.txt

4、案例解析

舉例如下：

1pwd?=?"G:\\a"
2file_path_list?=?[]
3for?path,dirs,files?in?os.walk(pwd):
4????for?file?in?files:
5????????file_path_list.append(os.path.join(pwd,file))
6print(file_path_list)

結果如下：

1['G:\\a\\aa.txt','G:\\a\\bb.xlsx','G:\\a\\cc.txt','G:\\a\\dd.docx']

5、怎么在一個列表中存放多個DataFrame數(shù)據(jù)。

1#?先使用如下代碼創(chuàng)建兩個DataFrame數(shù)據(jù)源。
2import?numpy?as?np
3xx?=?np.arange(15).reshape(5,3)
4yy?=?np.arange(1,16).reshape(5,3)
5xx?=?pd.DataFrame(xx,columns=["語文","數(shù)學","外語"])
6yy?=?pd.DataFrame(yy,columns=["語文","數(shù)學","外語"])
7print(xx)
8print(yy)

結果如下：

怎么講上述兩個DataFrame拼接在一起？

 1concat_list?=?[]
 2concat_list.append(xx)
 3concat_list.append(yy)
 4# pd.concat(list)中【默認axis=0】默認的是數(shù)據(jù)的縱向合并。
 5# pd.concat(list)括號中傳入的是一個DataFrame列表。
 6# ignore_list=True表示忽略原有索引，重新生成一組新的索引。
 7z?=?pd.concat(concat_list,ignore_list=True)
 8print(z)
 9
10#?或者直接可以寫成z?=?pd.concat([xx,yy],ignore_list=True)

結果如下：

二、多工作簿合并（一）

1、將多個Excel合并到一個Excel中（每個Excel中只有一個sheet表）

操作如下：

 1import?pandas?as?pd
 2import?os
 3pwd?=?"G:\\b"
 4df_list?=?[]
 5for?path,dirs,files?in?os.walk(pwd):
 6????for?file?in?files:
 7????????file_path?=?os.path.join(path,file)????????????????????????
 8????????df?=?pd.read_excel(file_path)?
 9????????df_list.append(df)
10result?=?pd.concat(df_list)
11print(result)
12result.to_excel('G:\\b\\result.xlsx',index=False)

結果如下：

三、多工作簿合并（二）

1、相關知識點講解

xlsxwrite的用法

 11）創(chuàng)建一個"工作簿"，此時里面會默認生成一個名叫"Sheet1"的Sheet表。
 2import?xlsxwriter
 3#?這一步相當于創(chuàng)建了一個新的"工作簿"；
 4#?"demo.xlsx"文件不存在，表示新建"工作簿"；
 5#?"demo.xlsx"文件存在，表示新建"工作簿"覆蓋原有的"工作簿"；
 6workbook?=?xlsxwriter.Workbook("demo.xlsx")
 7#?close是將"工作簿"保存關閉，這一步必須有。否則創(chuàng)建的文件無法顯示出來。
 8workbook.close()??
 92）創(chuàng)建一個"工作簿"并添加一個"工作表"，工作表命名為"2018年銷量"。
10import?xlsxwriter
11workbook?=?xlsxwriter.Workbook("cc.xlsx")
12worksheet?=?workbook.add_worksheet("2018年銷售量")
13workbook.close()???

結果如下：

 13）給"2018年銷售量"工作表創(chuàng)建一個表頭，向其中插入一條數(shù)據(jù)。
 2import?xlsxwriter
 3#?創(chuàng)建一個名為【demo.xlsx】工作簿；
 4workbook?=?xlsxwriter.Workbook("demo.xlsx")
 5#?創(chuàng)建一個名為【2018年銷售量】工作表；
 6worksheet?=?workbook.add_worksheet("2018年銷售量")
 7#?使用write_row方法，為【2018年銷售量】工作表，添加一個表頭；
 8headings?=?['產品','銷量',"單價"]?
 9worksheet.write_row('A1',headings)
10#?使用write方法，在【2018年銷售量】工作表中插入一條數(shù)據(jù)；
11# write語法格式：worksheet.write(行,列,數(shù)據(jù))?
12data?=?["蘋果",500,8.9]
13for?i?in?range(len(headings)):
14????worksheet.write(1,i,data[i])?
15workbook.close()

結果如下：

其他用法可以參考：https://www.cnblogs.com/brightbrother/p/8671077.html

xlrd的用法

 11）打開某一個存在的excel文件，返回給我們"xlrd.book.Book"工作簿對象；
 2#?這里所說的"打開"并不是實際意義上的打開，只是將該表加載到內存中打開。
 3#?我們并看不到"打開的這個效果"
 4#?以打開上述創(chuàng)建的"test.xlsx"文件為例；
 5import?xlrd?
 6file?=?r"G:\Jupyter\test.xlsx"
 7xlrd.open_workbook(file)
 8#?結果如下：
 90x29bb8e4eda0>
10
112）sheet_names()：獲取所有的sheet表表名，假如有多個sheet表，返回一個列表；
12import?xlrd
13file?=?r"G:\Jupyter\test.xlsx"
14fh?=?xlrd.open_workbook(file)
15fh.sheet_names()
16#?結果如下：
17['2018年銷售量',?'2019年銷售量']
18
193）sheets()方法：返回的是sheet表的對象列表。
20#?返回sheet表的對象列表
21fh.sheets()
22#?結果如下：
23[0x29bb8f07a90>,?0x29bb8ef1390>]
24
25#?可以利用索引，獲取每一個sheet表的對象
26fh.sheets()[0]
27結果是：0x29bb8f07a90>
28fh.sheets()[1]
29結果是：0x29bb8ef1390>
30
314）返回每一個sheet表的行數(shù)(nrows)?和?列數(shù)(ncols)；
32#?我們可以利用上述創(chuàng)建的sheet表對象，對每一個sheet表進行操作；
33fh.sheets()[0].nrows??#?結果是：4
34fh.sheets()[0].ncols??#?結果是：3
35
365）row_values(行數(shù))：獲取每一個sheet表中每一行的數(shù)據(jù)；
37sheet1?=?fh.sheets()[0]
38for?row?in?range(fh.sheets()[0].nrows):
39???value?=?sheet1.row_values(row)
40???print(value)

結果如下：

16）col_values(列數(shù))：獲取每一個sheet表中每一列的數(shù)據(jù)；
2sheet1?=?fh.sheets()[0]
3for?col?in?range(fh.sheets()[0].ncols):
4???value?=?sheet1.col_values(col)
5???print(value)

結果如下：

2、將多個Excel合并到一個Excel中（每個Excel中不只一個sheet表）

 1import?xlrd
 2import?xlsxwriter
 3import?os
 4
 5#?打開一個Excel文件，創(chuàng)建一個工作簿對象
 6def?open_xlsx(file):
 7????fh=xlrd.open_workbook(file)
 8????return?fh
 9
10#?獲取sheet表的個數(shù)
11def?get_sheet_num(fh):
12????x?=?len(fh.sheets())
13????return?x
14
15#?讀取文件內容并返回行內容
16def?get_file_content(file,shnum):
17????fh=open_xlsx(file)
18????table=fh.sheets()[shnum]
19????num=table.nrows
20????for?row?in?range(num):
21????????rdata=table.row_values(row)
22????????datavalue.append(rdata)
23????return?datavalue
24
25def?get_allxls(pwd):
26????allxls?=?[]
27????for?path,dirs,files?in?os.walk(pwd):
28????????for?file?in?files:
29????????????allxls.append(os.path.join(path,file))
30????return?allxls
31
32#?存儲所有讀取的結果
33datavalue?=?[]??
34pwd?=?"G:\\d"??????????????????????????
35for?fl?in?get_allxls(pwd):
36????fh?=?open_xlsx(fl)?????
37????x?=?get_sheet_num(fh)??
38????for?shnum?in?range(x):
39????????print("正在讀取文件："+str(fl)+"的第"+str(shnum)+"個sheet表的內容...")
40????????rvalue?=?get_file_content(fl,shnum)???
41
42#?定義最終合并后生成的新文件
43endfile?=?"G:\\d\\concat.xlsx"
44wb1=xlsxwriter.Workbook(endfile)
45#?創(chuàng)建一個sheet工作對象
46ws=wb1.add_worksheet()
47for?a?in?range(len(rvalue)):
48????for?b?in?range(len(rvalue[a])):
49????????c=rvalue[a][b]
50????????ws.write(a,b,c)
51wb1.close()
52print("文件合并完成")

【將上述代碼封裝后如下】

 1import?xlrd
 2import?xlsxwriter
 3import?os
 4
 5class?Xlrd():
 6????def?__init__(self,pwd):
 7????????self.datavalue?=?[]?
 8????????self.pwd?=?pwd?
 9
10????#?打開一個Excel文件，創(chuàng)建一個工作簿對象；
11????def?open_xlsx(self,fl):
12????????fh=xlrd.open_workbook(fl)
13????????return?fh
14
15????#?獲取sheet表的個數(shù)；
16????def?get_sheet_num(self,fh):
17????????x?=?len(fh.sheets())
18????????return?x
19
20????#?讀取不同工作簿中每一個sheet中的內容，并返回每行內容組成的列表；
21????def?get_file_content(self,file,shnum):
22????????fh?=?self.open_xlsx(file)
23????????table=fh.sheets()[shnum]
24????????num=table.nrows
25????????for?row?in?range(num):
26????????????rdata=table.row_values(row)
27????????????#?因為每一個sheet表都有一個表頭；
28????????????#?這里的判斷語句，把這個表頭去除掉；
29????????????#?然后在最后寫入數(shù)據(jù)的，添加上一個表頭，即可；
30????????????if?rdata?==?['姓名','性別','年齡','家庭住址']:
31????????????????pass
32????????????else:
33????????????????self.datavalue.append(rdata)
34????????return?self.datavalue
35
36????#?獲取xlsx文件的全路徑；
37????def?get_allxls(self):
38????????allxls?=?[]
39????????for?path,dirs,files?in?os.walk(self.pwd):
40????????????for?file?in?files:
41????????????????allxls.append(os.path.join(path,file))
42????????return?allxls
43
44????#?返回不同工作簿中，所有的sheet表的內容列表；
45????def?return_rvalue(self):
46????????for?fl?in?self.get_allxls():
47????????????fh?=?self.open_xlsx(fl)?????
48????????????x?=?self.get_sheet_num(fh)??
49????????????for?shnum?in?range(x):
50????????????????print("正在讀取文件："+str(fl)+"的第"+str(shnum)+"個sheet表的內容...")
51????????????????rvalue?=?self.get_file_content(fl,shnum)??
52????????return?rvalue
53
54class?Xlsxwriter():
55????def?__init__(self,endfile,rvalue):
56????????self.endfile?=?endfile
57????????self.rvalue?=?rvalue
58
59????def?save_data(self):
60????????wb1?=?xlsxwriter.Workbook(endfile)
61????????#?創(chuàng)建一個sheet工作對象；
62????????ws?=?wb1.add_worksheet("一年級(7)班")
63????????#?給文件添加表頭；
64????????ws?=?wb1.add_worksheet("2018年銷售量")
65????????headings?=?['姓名','性別','年齡','家庭住址']?
66????????for?a?in?range(len(self.rvalue)):
67????????????for?b?in?range(len(self.rvalue[a])):
68????????????????c?=?self.rvalue[a][b]
69????????????????#?因為給文件添加了表頭，因此，數(shù)據(jù)從下一行開始寫入；
70????????????????ws.write(a+1,b,c)
71????????wb1.close()
72????????print("文件合并完成")???
73
74pwd?=?"G:\\d"
75xl?=?Xlrd(pwd)
76rvalue?=?xl.return_rvalue()
77endfile?=?"G:\\d\\concat.xlsx"
78write?=?Xlsxwriter(endfile,rvalue)
79write.save_data()；

結果如下：

四、一個工作簿多sheet表合并。

1、將一個Excel表中的多個sheet表合并，并保存到同一個excel。

 1import?xlrd
 2import?pandas?as?pd
 3from?pandas?import?DataFrame
 4from?openpyxl?import?load_workbook
 5
 6excel_name?=?r"D:\pp.xlsx"
 7wb?=?xlrd.open_workbook(excel_name)
 8sheets?=?wb.sheet_names()
 9
10alldata?=?DataFrame()
11for?i?in?range(len(sheets)):
12????df?=?pd.read_excel(excel_name,?sheet_name=i,?index=False,?encoding='utf8')
13????alldata?=?alldata.append(df)???????
14
15writer?=?pd.ExcelWriter(r"C:\Users\Administrator\Desktop\score.xlsx",engine='openpyxl')
16book?=?load_workbook(writer.path)
17writer.book?=?book??
18#?必須要有上面這兩行，假如沒有這兩行，則會刪去其余的sheet表，只保留最終合并的sheet表
19
20alldata.to_excel(excel_writer=writer,sheet_name="ALLDATA")
21writer.save()
22writer.close()

結果如下：

五、一表拆分(按照表中某一列進行拆分)

1、將一個Excel表，按某一列拆分成多張表。

 1import?pandas?as?pd
 2import?xlsxwriter
 3data=pd.read_excel(r"C:\Users\Administrator\Desktop\chaifen.xlsx",encoding='gbk')
 4
 5area_list=list(set(data['店鋪']))
 6
 7writer=pd.ExcelWriter(r"C:\Users\Administrator\Desktop\拆好的表1.xlsx",engine='xlsxwriter')
 8data.to_excel(writer,sheet_name="總表",index=False)
 9
10for?j?in?area_list:
11????df=data[data['店鋪']==j]
12????df.to_excel(writer,sheet_name=j,index=False)
13
14writer.save()??#一定要加上這句代碼，“拆好的表”才會顯示出來

結果如下：

推薦閱讀

*《學習 Python 有這些資源足以！》
*《學習 Python 的 14 張思維導圖~》
*《改善 Python 程序的91個建議》
*《Linux 環(huán)境下安裝 Python3 的操作方法》
*《Python 運維中20個常用的庫和模塊》
*《做到這 20 條，Python 程序性能輕松翻倍！》
*《一篇了解 Python 爬蟲技術方方面面》
*《常見的 25 個 Python 面試問答》