標準入力からキーとバリューを交互に格納したリストを作成したい。

Question

#やりたいこと
標準入力から取得したデータで、キーとバリューを交互に格納したリストを作成したい。

#標準入力の値
noda 1000
ando 500

#書いたコード

import sys

lst = [s.rstrip().split(" ") for s in sys.stdin]
print(lst)

#期待する出力
[['noda', 1000 ], ['ando', 500 ]]

#現在の出力
[['noda', '1000'], ['ando', '500']]

各要素の2つ目を数値型で取得したいのですが、
どのように書き直せばいいのか分かりません・・・
よろしくお願いいたします。

Answer 1

内包表記とはちょっと相性が悪いです。
適当に関数を用意しておくのが分かりやすく／改良しやすいように思います。

def parse_record(line):
    k, v = line.split()
    return k, int(v)

lst = [list(parse_record(line)) for line in map(str.strip, sys.stdin)]

内包表記で頑張るならこんな感じでしょうか。

lst = [
    [k, int(v)] for k, v in map(lambda l: l.strip().split(), sys.stdin)
]

もうちょっと簡潔に書けるかもしれないですが。

あるいは

『整数に直せるものは全部直す』という仕様ならば。

lst = [
    [int(e) if e.isdigit() else e for e in line.split()]
    for line in map(str.strip, sys.stdin)
]

読みづらいですね。条件演算子を使っています。

Answer 2

>>> import sys
>>> [(lambda a,b:[a, int(b)])(*s.rstrip().split(" ")) for s in sys.stdin]
noda 1000
ando 500
[['noda', 1000], ['ando', 500]]

lambdaがあるから無駄な名前がいらない（外部リンク）というパターンですね。