技術とメンタル

技術的なことと考えていることのアウトプット

python3の辞書で標準入力からキーを得て値を出力する

標準入力からキーを受け取り、辞書を参照して値を標準出力する方法のメモ。

まず、

mydict  #あらかじめ用意しておいた辞書

があるとすると、

print(mydict[input()])

と記述すれば、そのキーに合った値を出力してくれる。




以下、入力、プログラム、出力のサンプルを載せる。

入力は、1行目にキーと値の組み合わせの数を
2行目以降はキーと値の組み合わせが続くとする。
そして、下2行に実際に辞書から検索してその値を表示したいキーを追加した。

3
aaa 1000
bbb 2000
ccc 3000
aaa
ccc

その時、プログラムは

N = int(input())

#init
mylist=[]

#make list
for i in range(N):
    mylist.append(input().split())

#make dictionary from list    
mydict=dict(mylist)

#output
for i in range(N):
    tmp = input()
    if tmp in mydict:
        print(tmp+"=",end="")
        print(mydict[tmp])
    else:
        print("Not found")


出力は

aaa=1000
Not found
ccc=3000

となる。