Excelファイルに入力した番号と対応した名前を出力させたい

Question

name_list = []
wb = openpyxl.load_workbook("C:\user\name.xlsx")
ws = wb["名前"]
for row in ws.iter_rows(min_row=2):
    values = []
    for col in row:
        values.append(col.value)
    name_list.append(tuple(values))

openpyxlを使って上記のexcelファイルを取り込んだ。

segments = [1,2,3,6]

そして上記のようなリストを作成したあとに、segmentsに入れた番号順で名前を表示させ、もし入力した番号に対応する名前がなかったらそのまま番号を出力させるようにしたいのですが、うまくできませんでした。

目的の出力
['佐藤','高橋','田中','6']

---

試したこと

masters = []
for s in segments:    
    if s != name_list[0]:
        masters.append(s)
    for d in name_list:    
        if s == d[0]:
            masters.append(d[1])

上記のプログラムを書いたが['1','佐藤','2','高橋','3','田中','6']となってしまいどのようにすればいいかわかりませんでした。

Answer 1

name_listがリストだと処理が大変なので、辞書型(name_dict)にしました。
あと、xlsx読み込み時にfor row in ws.iter_rows(min_row=2):となっていて、1行目が読まれていなかったので勝手にmin_row=1としています。必要なら戻してください。

import openpyxl

name_dict = {}
wb = openpyxl.load_workbook("C:\user\name.xlsx")
ws = wb["Sheet1"]
for row in ws.iter_rows(min_row=1):
    values = []
    for col in row:
        values.append(col.value)
    name_dict[values[0]] = values[1]
print(name_dict)
# {1: '佐藤', 2: '高橋', 3: '田中', 4: '鈴木', 5: '山田'}

masters = []
segments = [1,2,3,6]
for s in segments:
    if s in name_dict:
        masters.append(name_dict[s])
    else:
        masters.append(s)
print(masters)
# ['佐藤', '高橋', '田中', 6]