pyhonでexcelファイルを読み込み列幅の自動調整をしたい

Question

pyhonでexcelファイルを読み込み、列幅を自動調整して上書き保存するためのプログラムを作成しています。現状、下記プログラムを作成し実行してみました。

import pandas as pd
import openpyxl as xl
path = 'AAA'

# ディレクトリ階層+中身 が各列に入る2次元リストを作る
# [['AAA'],
#  ['AAA', 'test1.xlsx'],
#  ['AAA', 'BBB'],
#  ['AAA', 'BBB', 'test2.xlsx'],
#  ...] 
pathlist = []
for curDir, dirs, files in os.walk(path):
    leaf = curDir.split('/')
    pathlist.append(leaf)
    for a_dir in dirs:
        pathlist.append(leaf + [a_dir])
    for a_file in files:
        pathlist.append(leaf + [a_file])

# 作った2次元リストをデータフレームにする
path_df = pd.DataFrame(pathlist)

# 重複削除
path_df = path_df.drop_duplicates()  
print(path_df)
    
path_df.to_excel('sample.xlsx')

# set input file name
inputfile = 'sample.xlsx'

# read input xlsx
wb1 = xl.load_workbook(filename=inputfile)
ws1 = wb1.worksheets[0]

# set column width
for col in ws1.columns:
    max_length = 0
    column = col[0].column

    for cell in col:
        if len(str(cell.value)) > max_length:
            max_length = len(str(cell.value))

    adjusted_width = (max_length + 2) * 1.2
    ws1.column_dimensions[column].width = adjusted_width

# save xlsx file
wb1.save(inputfile)

実行すると、下記エラーが発生します。
File "C:\Users\NEC\anaconda3\lib\site-packages\openpyxl\descriptors\base.py", line 42, in set
raise TypeError('expected ' + str(self.expected_type))

TypeError: expected <class 'str'>

このエラーを回避して自動調整するには、私が作成した上記プログラムは問題なくて、
base.pyのプログラムの中身を修正する必要があるのでしょうか。

どなたかお分かりになる方いましたら、ご教授お願いいたします。

Comments

[kuma_kuma_]

> if len(str(cell.value)) > max_length:

if len(str(cell.value + "")) > max_length:

[meg_]

エラー発生箇所はどこですか？

[ot1982]

1行ずつ実行していくと、下記コードのところで発生しています。
ws1.column_dimensions[column].width = adjusted_width

Answer 1

column_dimensionsの[]内には、列名の文字列('A'とか)を指定します。
提示されたコードでは列番号(1~)の数値を指定しているのでエラーとなります。

列名の文字列をcol[0].column_letterで取得して指定すればエラーは解消されます。

-    ws1.column_dimensions[column].width = adjusted_width
+    ws1.column_dimensions[col[0].column_letter].width = adjusted_width

補足:
なぜ、検索してコピペしたコードが動かなかったのかというと、openpyxl 3以降でcolumn_dimensionsの添字が列番号の数値から列名の文字列に変更されたから、ですね。参考にしたサイトは、古いバージョンで動作するコードだったのでしょう。

Answer 2

column = col[0].columnを下記のように変えたら回避できました。
column = col[0].column_letter
なぜ変えたらできたのか理由がわからないので、引き続き調べてみようと思います。

Comments

[Daregada]

いや、回答読んでくれよ

[ot1982]

すみません、回答していただいていたのに気づいておらず、先ほど確認しました。
column_dimensionsの[]内は、数値を指定しては駄目なんですね。勉強になります。
ご丁寧に回答いただきましてありがとうございました！

[Daregada]

あなたがおそらく参考にしたであろうサイト
https://qiita.com/github-nakasho/items/62eaba26d5ee4f13a8ac
や、おそらくその基になった情報、
https://stackoverflow.com/questions/39529662/python-automatically-adjust-width-of-an-excel-files-columns
には、列番号を指定した例が書かれていますね。あなたのコードの後半はこれのコピペですね。

[ot1982]

そうです、プログラムの作り方が分からなかったので、できそうなコードを見つけて、うまくいくか試していました。