for文の組み合わせについて

Question

wb = openpyxl.load_workbook(excelname)
sheet1 = wb[sheet_name1]
sheet2 = wb[sheet_name2]
sheet3 = wb[sheet_name3]
sheet4 = wb[sheet_name4]

ss1 = [sheet1, sheet4]
ss2 = [sheet2, sheet3]

hiddenlist1 = ['F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']
hiddenlist2 = ['G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']

#sheet1のみだったら
for col1 in hiddenlist1:
    sheet1.column_dimensions[col1].hidden = True

wb.save(excelname)

sheet1,sheet4はhiddenlist1の列を非表示にする。
sheet2,sheet3はhiddenlist2の列を非表示にしたいのですが、
うまくfor文をまとめることができず困っております。
初心者な質問で申し訳ございませんがご教授いただけると助かります。
よろしくお願いいたします。

#追記

if __name__ == '__main__':
    wb = openpyxl.load_workbook(excelname)
    ss1 = [sheet1, sheet4]
    ss2 = [sheet2, sheet3]
    ss_all = [sheet1, sheet2, sheet3, sheet4]

    hiddenlist1 = ['F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']
    hiddenlist2 = ['G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']

    for ws in ss_all:
        if ws in ss1:
            for col in hiddenlist1:
                ws.column_dimensions[col].hidden = True
                pass

        elif ws in ss2:
            for col in hiddenlist2:
                ws.column_dimensions[col].hidden = True
                pass

wb.save(excelname)

上記のように書いてみました、
エラーは出ておりませんが一つのシートも列が非表示になりませんでした。

追記2
sheet1などの変数に関しましては下記のようになっています。
AAAはエクセルのシート名です。shhe2なども同様に全て定義しています。

sheet_name1 = 'AAA'
sheet1 = wb[sheet_name1]

Answer 1

関数に分けると見通しが良くなります。

def hide_cells(sheet, hiddenlist):
    for col1 in hiddenlist:
        sheet.column_dimensions[col1].hidden = True

def hide_cells_in_sheets(sheets, hiddenlist):
    for sheet in sheets:
        hide_cells(sheet, hiddenlist)

hide_cells_in_sheets(ss1, hiddenlist1)
hide_cells_in_sheets(ss2, hiddenlist2)

wb.save(excelname)

と書くと、あとで修正する場合も楽になります。

hiddenlistがもっと増えてきたら、さらにこれをまとめる関数を作った方がわかりやすいですが、二個ならこれぐらいで十分でしょう。

Comments

[jazzmaster]

回答ありがとうございます。想定通り動きました！とても勉強になりました。

Answer 2

ご質問内容がfor文をまとめることであれば、以下のようにしてみてください。
(openpyxl: Excel操作のライブラリ?については詳細が分からないので試していません。)

ポイント

• ワークブックを開いた後、シート名を各変数に代入している箇所でsheet_name1..4を設定すると、

wb.sheetnamesが取れると思います。(実際に試したわけではないので、代わりにsheet変数を
追加しました。)

• forを追加するのは、#sheet1のみだったら　の後で、sheetを総当たりしつつ、

  • 該当シート名がss1に含まれていれば、ss1の処理を実施。
  • 該当シート名がss2に含まれていれば、ss2の処理を実施。

• wb['sheetname']でシートが取得できるそうなので、wb[ws]として取る感じです。

if __name__ == '__main__':
    # wb = openpyxl.load_workbook(excelname)
    # sheet1 = wb[sheet_name1]
    # sheet2 = wb[sheet_name2]
    # sheet3 = wb[sheet_name3]
    # sheet4 = wb[sheet_name4]
    # sheets = wb.sheetnames
    sheets = ["sheet1", "sheet2", "sheet3", "sheet4"]

    ss1 = ["sheet1", "sheet4"]
    ss2 = ["sheet2", "sheet3"]

    hiddenlist1 = ['F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']
    hiddenlist2 = ['G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']

    # ------- ここでシートをループする ---------
    for ws in sheets:
        if ws in ss1:
            print(ws)
            print(hiddenlist1)
            for col in hiddenlist1:
                # wb[ws].column_dimensions[col].hidden = True
                pass
        
        if ws in ss2:
            print(ws)
            print(hiddenlist2)
            for col in hiddenlist2:
                # wb[ws].column_dimensions[col].hidden = True
                pass

    # wb.save(excelname)

追記(やってみた)

ご質問はfor文をまとめる方法で合っていますか?
これだと成功しました。

import openpyxl

if __name__ == '__main__':
    sheet_name1 = "Sheet1"
    sheet_name2 = "Sheet2"
    sheet_name3 = "Sheet3"
    sheet_name4 = "Sheet4"
    wb = openpyxl.load_workbook("Sample.xlsx")

    ss1 = [sheet_name1, sheet_name4]
    ss2 = [sheet_name2, sheet_name3]

    hiddenlist1 = ['F','G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']
    hiddenlist2 = ['G','H','I','J','K','L','M','N','O','P','Q','R','S','T','U']

    for ws in wb.sheetnames:
        if ws in ss1:
            print(f"ss1: {ws}")
            # for col in hiddenlist1:
            #     ws.column_dimensions[col].hidden = True

        elif ws in ss2:
            print(f"ss2 {ws}")
            # for col in hiddenlist2:
            #     ws.column_dimensions[col].hidden = True

# wb.save(excelname)

実行結果

ss1: Sheet1
ss2 Sheet2
ss2 Sheet3
ss1: Sheet4

Comments

[jazzmaster]

回答ありがとうございます。エラーは出なかったのですが列が非表示にはなりませんでした。
私が書いたものを追記として記載させていただきます。

[rtazaki]

ss1, ss2のリストに入っているのは何でしょうか。
wb[sheet_name1]の戻り値であればシート名ではなく、ワークシートそのものだと思います。
if ws in ss1:でss1と存在チェックしたときに、オブジェクトで一致が見れてないのではと思います。

なお、passは、for文で字下げしたときに処理なしだとpythonでは文法エラーになるため、
passを追加しただけですので、ws.column_dimensions[col].hidden = Trueの後のpassは2つとも不要です。

[jazzmaster]

追記させていただきました。

[rtazaki]

追記しました。
ご自身の実際のコードにprint入れるか、breakpointを入れてデバッグしてみてください。
if ws in ss1:の箇所で想定外の動きをしていると思います。

[jazzmaster]

回答ありがとうございます！こちらも明日試してみたいと思います。

[rtazaki]

想定通り動いたのであれば良かったです。