Python3 複数行データに文字と数値が混在した場合でも文字と数値として1度に入力することが可能なのか？

Question

オンラインの問題などでよくあるパターンなのですが、
1行目にN=その後にあるデータの行数
2行目以降は空白区切りで複数個のデータを含む行がN行など

sample = [list(map(int, input().split())) for i in range(N)]

例えば入力データが全て整数であれば上記のように**map(int, input().split())**を使えば良いのですが、
文字と数値が混在してる場合このままで1度に入力処理させると文字を含むので当然エラーとなります。

####知りたいこと
複数行データに文字と数値が混在した場合でも文字と数値として1度に入力することが可能なのか？
そのような方法が無いという回答でも構いません。
ヒントやアドバイスあればご教示頂けると助かります。

####やったこと
プログラミング(競プロ)に便利な Python3 用法集　入力編
上記サイトを参考にしてもやはり複数行で文字と数値の混在パターンのサンプルコードはありませんでした。
現状は一旦文字列としてリストに取り込み、その後for文で全要素をint()やfloat()で処理して上書きしています。

入力データ
3
a 1.23
b 2.34
c 3.45

N = int(input().rstrip())
src = [list(input().split()) for _ in range(N)]
print(src)
for row in src:
    row[1] = float(row[1])
print(src)

出力（リストの中身確認）
[['a', '1.23'], ['b', '2.34'], ['c', '3.45']]
[['a', 1.23], ['b', 2.34], ['c', 3.45]]

####環境
Python 3.7.0 or 3.6.6

Answer 1

こういう強引な方法も。

def cast_data(datum):
    assert isinstance(datum, str)
    
    if datum.isdigit():
        return int(datum)
    try:
        return float(datum)
    except ValueError:
        return datum


N = int(input())
data = [
    [
        cast_data(datum) for datum in input().split()
    ]
    for _ in range(N)
]

print(data)

実行結果 Wandbox

[['a', 1.23], ['b', 2.34], ['c', 3.45]]

汎用性はともかく実行効率は良くないと思います。計測しないと何とも言えませんが。

Comments

[opyon]

ありがとうございます。
ちょっと読み解くのに時間がかかりそうですが今後の参考になりそうな予感がします。

Answer 2

それ用の関数を定義すれば簡単です。

def f(s):
    lst = s.split()
    return lst[0], float(lst[1])

sample = [f(input()) for i in range(N)]

その気になればlambdaで書けます。メリットは特にありません。

sample = [(lambda s_lst: (s_lst[0], float(s_lst[1])))(input().split()) for i in range(N)]

Comments

[opyon]

ありがとうございます。どちらも目から鱗です><.
因みにこれを検索しようとするとどのようなキーワードが思いつきますか？
自分なりには検索してみたつもりだったのですが辿り着けずに質問させて頂きました。

[hayataka2049]

キーワードは、思いつきません
お力になれなくてすみません

[opyon]

いえいえとんでもないです。ありがとうございました。
頂いたコードをある程度パターン化しておいて問題に挑みたいと思います。

[hayataka2049]

こういう部分は、多少泥臭く書いてもべつに良いような気がします。慣れてきたらスマートな書き方を模索してみる程度で
あとはatcoderとかなら、同じ言語で書いている他の人のコードをのぞき見すると色々ヒントが得られるかもしれません

[opyon]

for文を何度も使った泥臭い方法はなんとかなることもあるのですが。
atcoderはまだほとんど使ったことがないですが、なるほど確かに覗くのが近道かもしれませんね。

Answer 3

scanf を使いたいということでしょうか？

参考情報

• Pythonにおける標準入力について(scanfみたいに使いたい)

https://teratail.com/questions/57605

• pythonでscanf的なことをする

https://hayataka2049.hatenablog.jp/entry/2018/04/10/155957

scanf は、 pip3 install scanf をすれば、後はこんなふうに使えます。

Comments

[opyon]

>以上を読んで、それでもscanf()を使いたい、自分にはscanf()しかない、と思うのなら、
>モジュールをインストールすればできます。
>PyPI: scanf
ありがとうございます。
質問した時はjavaみたいにscanfあればいいなと思っていました。（よく分かりましたね）
どうしてもscanfを使いたいわけではないので教えていただいた別の方法を使うつもりです。