python(openpyxl) zipファイル(csv)→xlsxに処理、変換するには

Question

プログラミング初学者です。拙い質問かもしれませんが宜しくお願いします。
現在、Pythonの勉強を兼ねてopenpyxlでのエクセルファイルの処理を勉強しています。
やりたいこととしては、
・zipファイル内にある複数のcavファイルをそれぞれシートに変え、xlsx形式で保存したい
と考えています。

少しググってみた結果、zipファイルの中身を確認する方法と
xlsx形式で保存する方法はわかりましたが、
それぞれを連携させる方法がわからないので、ご教授頂きたいです。

# zipファイルを読み込む
import zipfile
zf = zipfile.ZipFile('hogehoge.zip')
lst = zf.namelist()
print(lst)
# ファイル名だけ渡されても、どうしろと。

#フォルダのcsvをひとつのエクセルに
import os
import glob
from pathlib import Path

import openpyxl
import csv

csvfiles = glob.glob("root/*.csv", recursive=False)
wb = openpyxl.Workbook()
for file in csvfiles:
    wb.create_sheet(os.path.splitext(os.path.basename(file))[0])
    #よくわからんけど空欄のシートができる
    wb.active = wb.sheetnames.index(os.path.splitext(os.path.basename(file))[0])
    ws = wb.active
    with open(file, encoding="shift-jis") as f:
        reader = csv.reader(f, delimiter=',')
        for row in reader:
            ws.append(row)
wb.save("完成.xlsx")

Comments

[t_obara]

次は、zipファイルに含まれているファイルを取り出す方法を調査すれば良いのではないでしょうか。

Answer 1

まず、zf.openを使ってzfからcsvをstrとして取り出します。

>>> import zipfile
>>> import csv
>>> import io
>>> 
>>> zf = zipfile.ZipFile('hogehoge.zip')
>>> lst = zf.namelist()
>>> print(lst)
['test.csv']
>>>
>>> with zf.open('test.csv') as f:
...     csv_bytes = f.read()
...     csv_str = csv_bytes.decode(encoding='cp932')
...
>>> print(type(csv_str))
<class 'str'>
>>> print(csv_str)
date,val
202001,d
202001,f
202001,g
202001,h
202001,i

使い方は、ZipFile オブジェクトを読んでください。

次に、io.StringIOを使って、csv_strを指定してcsv.readerを作ります。

>>> with io.StringIO(csv_str) as f:
...     reader = csv.reader(f, delimiter=',')
...     for row in reader:
...         print(row)
...
['date', 'val']
['202001', 'd']
['202001', 'f']
['202001', 'g']
['202001', 'h']
['202001', 'i']

使い方は、class io.StringIOを読んでください。

あとは、ご自身のコードに組み込んでください。