python3 readlineを使った、表示方法について

Question

最近、python3を始めたものです。
標準入力で入力した複数行の文字を元に画像を出力するプログラムを書いています。
しかし、エラーが出てしまい、分かりません。
また、他のアプリ（visual studio 2017）のデバック機能を使って解決しようとしましたが、今度は複数行の標準入力の終了の仕方が分からず出来ませんでした。この事についても、どうか回答の程よろしくお願いいたします。

#プログラム
画像用辞書
items_img = {
"剣" : "http://paiza.jp/learning/images/sword.png",
"盾" : "http://paiza.jp/learning/images/shield.png",
"回復薬" : "http://paiza.jp/learning/images/potion.png",
"クリスタル" : "http://paiza.jp/learning/images/crystal.png"
}

import sys
for i in sys.stdin.readlines():
    print("<img src='"+items_img[i]+"'>")

#入力　

6
回復薬
盾
クリスタル
クリスタル
剣
剣


※標準入力で与えられるアイテム名は、items_imgに無いアイテムは
出てこないものとします

#エラーメッセージ
Traceback (most recent call last):
  File "Main.py", line 13, in <module>
    print("<img src='"+items_img[i]+"'>")
KeyError: '6\n'

(Exit status: 1)

Answer 1

受け取るアイテム数を変数に代入

item_cnt = int(input())

アイテムを受け取る

while item_cnt > 0:
item = input()
print("<img src = '" + items_img[item] + "'>")
item_cnt -= 1

Comments

[bonji]

回答ありがとうございます。教えていただいて方法で実践したみたら、上手く行きました。
ありがとうございます。

Answer 2

対応すべき点は二つです。

一行目を読まないようにする
スライスを利用します。sys.stdin.readlines()[1:]

改行文字\nを排除
文字列lineに対しては、line.rstrip()で文末の余分な空白文字を削除できます。

---

複数行の標準入力の終了の仕方が分からず出来ませんでした

明示的にEOFを送出すれば良いです。
Windowsの場合 Ctrl+Z で、Mac／Linux系の場合 Ctrl+D で入力できます。

ただし、今回のバグの場合はエラーメッセージをしっかり読めば充分対応できるはずです。

コードの書き方について

teratailには、コードを見やすく表示する機能があります。
質問編集画面を開き、コードを選択した状態で<code>ボタンを押してください。

特にPythonの場合、インデントが崩れるとコードの意味が変わってしまいます。

Comments

[bonji]

回答とコードの書き方についてのアドバイスありがとうございました。早速、直しました！
①「一行目を読まないようにする」の[1:]は標準入力の（0からはじまる）読み取り開始位置を示しているのですよね？合ってますか？
②「改行文字\nを削除」するためのrstrip()をコードの何処に挿入したらいいか分かりません。稚拙な質問ですが回答の程よろしくお願いします。

[LouiS0616]

① sys.stdin.readlinesはリストを返しますので、スライスを用いてリストの第一要素以降を取り出しています。『Python スライス』などとググってみましょう。

② 文字列lineを取得した直後に line = line.rstrip() すれば良いです。なお、文字列 line は質問者さんのプログラム内の変数 i に相当します。

[bonji]

回答ありがとうございました。おかげさまで無事に実行することが出来ました。