Python 入力桁が変化する場合の値の取得

* 演算子 (iterable unpacking operator)を使う

例

1 2
2 1 20
1 1

みたいな入力で、３回ループさせたいのがわかっている場合

for i in range(q):
    t, x, *y = map(int, input().split())

# 3個目の入力がある場合のみ、*yに値がセットされる

qiita.com