Hello,我是小張,大家好久不見~
今天文章介紹一個實戰案例,與自動化辦公相關;案例思想是源于前兩天幫讀者做了一個 demo ,需求大致將一上百個 word 中表格內容提取出來(所有word 中表格樣式一樣),把提取到的內容自動存入 Excel 中
word 中表格形式如下
目前含有數個上面形式的 word 文檔需要整理,目標是利用 python 自動生成下面形式 excel 表格
正式案例講解之前,先看一下轉換效果,腳本先把指定文件夾下的 doc 文件轉化為 docx ,隨后自動生成一個 excel 表格,表格內中即為所有 word 中的內容
涉及的庫
本案例中用到的 Python 庫有以下幾個
python-docx
pandas
os
pywin32
doc 轉化為 docx
本案例中 word 中表格內容的提取用到的是 python-docx 庫,關于 python-docx 一些基礎用法可以參考
,
word 文檔有時是以 doc 類型保存的, python-docx 只能處理 docx 文件類型,在提取表格內容之前,需進行一次文件類型格式轉換:把 doc 批量轉化為 docx;
doc 轉 docx 最簡單的方式 通過Office 中 word 組件打開 doc 文件,然后手動保存為 docx 文件,對于單個文檔這個方法還行,文檔數量達到上百個的話還用這種方法就有點煩了,
這里介紹一個 python 庫 pywin32 來幫助我們解決這個問題,pywin32 作為擴展模塊, 里面封裝了大量 Windows API 函數,例如調用 Office 等應用組件、刪除指定文件、獲取鼠標坐標等等
利用 pywin32 控制Office 中 Word 組件自動完成 打開、保存
操作,把所有 doc 文件類型轉化為 docx 文件類型,步驟分為以下三步:
1,建立一個 word 組件
from win32com import client as wc
word = wc.Dispatch('Word.Application')</pre>
2,打開 word 文件
doc = word.Documents.Open(path)</pre>
3,保存關閉
doc.SaveAs(save_path,12, False, "", True, "", False, False, False, False)
doc.Close()
完整代碼
path_list = os.listdir(path)
doc_list = [os.path.join(path,str(i)) for i in path_list if str(i).endswith('doc')]
word = wc.Dispatch('Word.Application')
print(doc_list)
for path in doc_list:
print(path)
save_path = str(path).replace('doc','docx')
doc = word.Documents.Open(path)
doc.SaveAs(save_path,12, False, "", True, "", False, False, False, False)
doc.Close()
print('{} Save sucessfully '.format(save_path))
word.Quit()
docx 庫提取單個表格內容
在批量操作之前,首先需要搞定單個表格中的內容,只要我們搞定了單個 word,剩下的加一個遞歸即可
用 docx 庫對 word 中表格內容提取,主要用到 Table、rows、cells 等對象
Table 表示表格,rows 表示表格中行列表,以迭代器形式存在;cells 表示單元格列表,也是以迭代器形式
操作之前,需了解下面幾個基礎函數
通過 Document 函數讀取文件路徑,返回一個 Document 對象
Document.tables 可返回 word 中的表格列表;
table.rows 返回表格中的行列表;
row.cells 返回該行中含有的單元格列表;
cell.text 返回該單元格中文本信息
了解了上面內容之后,接下來的操作思路就比較清晰了;word 表格中文本信息可以通過兩個 for 循環來完成:第一個 for 循環獲取表格中所有行對象,第二個 for 循環定位每一行的單元格,借助 cell.text 獲取單元格文本內容;
用代碼試一下這個思路是否可行
document = docx.Document(doc_path)
for table in document.tables:
for row_index,row in enumerate(table.rows):
for col_index,cell in enumerate(row.cells):
print(' pos index is ({},{})'.format(row_index,col_index))
print('cell text is {}'.format(cell.text))
會發現,最終提取到的內容是有重復的,,,
出現上面原因,是由于單元格合并問題,例如下面表格的單元格 是 合并了 (1,1)->(1,5)
,docx 庫在處理這類合并單元格
時并沒有當成一個,而是以單個形式進行處理,因此 for 迭代時 (1,1)->(1,5)
單元格返回了五個,每一個單元格文本信息都返回 是
面對以上文本重復問題,需要添加一個去重機制, 姓名、性別、年齡...學歷學位
等字段作為列名 col_keys,后面王五、女、37、... 學士
等作為col_values,提取時設定一個索引,偶數為 col_keys, 奇數為 col_vaues ;
代碼重構后如下:
document = docx.Document(doc_path)
col_keys = [] # 獲取列名
col_values = [] # 獲取列值
index_num = 0
# 添加一個去重機制
fore_str = ''
for table in document.tables:
for row_index,row in enumerate(table.rows):
for col_index,cell in enumerate(row.cells):
if fore_str != cell.text:
if index_num % 2==0:
col_keys.append(cell.text)
else:
col_values.append(cell.text)
fore_str = cell.text
index_num +=1
print(f'col keys is {col_keys}')
print(f'col values is {col_values}')</pre>
最終提取后的效果如下
批量 word 提取,保存至 csv 文件中
能夠處理單個 word 文件之后,一個遞歸即可提取到所有 word 文本表格內容,最后利用 pandas 把獲取到的數據寫入到 csv 文件即可!
def GetData_frompath(doc_path):
document = docx.Document(doc_path)
col_keys = [] # 獲取列名
col_values = [] # 獲取列值
index_num = 0
# 添加一個去重機制
fore_str = ''
for table in document.tables:
for row_index,row in enumerate(table.rows):
for col_index,cell in enumerate(row.cells):
if fore_str != cell.text:
if index_num % 2==0:
col_keys.append(cell.text)
else:
col_values.append(cell.text)
fore_str = cell.text
index_num +=1
return col_keys,col_values
pd_data = []
for index,single_path in enumerate(wordlist_path):
col_names,col_values = GetData_frompath(single_path)
if index == 0:
pd_data.append(col_names)
pd_data.append(col_values)
else:
pd_data.append(col_values)
df = pd.DataFrame(pd_data)
df.to_csv(word_paths+'/result.csv', encoding='utf_8_sig',index=False)</pre>
證件號、身份證號格式
打開生成的 csv 文件會發現聯系方式、身份證號
兩欄的數字格式是以數值存儲,不是我們想要的類型,想要完整展示,需存儲之前把數值轉化為文本
解決方法,找到所在的單元格,前面元素前面加一個 ’\t‘ 制表符即可
col_values[7] = '\t'+col_values[7]
col_values[8] = '\t'+col_values[8]</pre>
源碼獲取
本案例中用到的 源碼數據獲取方式,關注微信公號:小張Python,在公號后臺回復關鍵字:210328 即可!
小結
本案例中只用到了 docx 庫中的一部分方法,主要涉及到了 word 中 Table 的基本操作,對于一些從事文職工作的同學來說日常工作中可能會遇到上面相似問題,因此特意分享在這里,希望能夠對大家有所幫助
好了,以上就是本篇文章的全部內容了,最后感謝大家的閱讀,我們下期見!