openpyxlで内容を上書きしてから取り込みたい

Question

お世話になります。
よろしくお願いいたします。

openpyxlで取り込んだエクセルのシートを
ブラウザで閲覧できるビューワーをpython3で作ってます。

以下が取り込みの処理になります。

import openpyxl

wb = openpyxl.load_workbook("sample.xlsx", data_only=True)
ws = wb["Sheet1"]

    for r in ws.rows:
        if r[0].row == 1:
            header_cells = r
        else:
            row_dict = {}
            for k,v in zip(header_cells,r):
                row_dict[k.value] = v.value

シートのヘッダーと該当セルを辞書化してhtmlに渡してます。
その際他の列の内容によって、内容を上書きして取り込みたいセルがありますが
どのように書けばよいのか、悩んでいます。

	　A	B
1	山	0	
2	海	0
3	山	0
4	海	1	#B列に1が入ってる場合、A列の"海"を"川"として取り込みたい

if B == 1:
A = "川"
みたいな単純な書き方ではダメで、しばらく考えあぐねています。
ご教示いただけますと幸いです。
どうぞよろしくお願いいたします。

Comments

[meg_]

実証可能なコードを提示いただくと回答がつきやすくなるかと思います。

[tonk]

追記いたしました。よろしくお願いいたします。

[meg_]

追記されたコードはforのところで「unexpcted indent」のエラーが出ますよ。

Answer 1

pandasを使ってDataFrameで変換してしまった方が楽ではないでしょうか。

import openpyxl
import pandas as pd

wb = openpyxl.load_workbook("data\sample.xlsx")
ws = wb.active

df = pd.DataFrame(ws.values, columns=["col1" ,"col2"])

#col2列が1のcol1列を'川'に書き換える
df.loc[df.col2 == 1, "col1"] = '川'

print(df[['col1']])

追記
pandasの方がSQLっぽく操作できて便利なので書いてみましたが、
それを使わずVBA的に書くならこうなりますかね。
辞書形式にしたいように見えたので、Excelの行数をキーにしてあります。

import openpyxl

wb = openpyxl.load_workbook("data\sample.xlsx")
ws = wb.active

row_dict = {}
for r in ws.rows:
    if r[1].value == 1:
        r[0].value = '川'
    row_dict[r[0].row] = r[0].value

print(row_dict)

Comments

[tonk]

とてもシンプルに書けるんですね。
ご提案ありがとうございます。検討してみたいと思います。
やはりopenpyxlで標記のことを実現するのは困難なのでしょうかね。

[tonk]

改めてのご回答ありがとうございました。
実現したいことをネットで都度調べながら書いてるので
リストで取り出せることも分かってませんでした。
ありがとうございました。

Answer 2

簡単な例です。

for i in range(1, 10):
    if ws.cell(row=i, column=2).value == 1:
        if ws.cell(row=i, column=1).value == '海':
            print('川') #ここに実際の処理を書く

Comments

[tonk]

ご回答ありがとうございました。
様々な実現の仕方を知ることができて大変勉強になりました。