Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
184 views
in Technique[技术] by (71.8m points)

python - list comprehension with input().split()

I have to read N then N tuples of two numbers as shown in the example below:

3
1 85
2 91
3 73

After that I want to sort them based on the second element, with ties broken by the order they came into input. To do that I wanted to save a tuple with 3 elements but I can't figure out how to put that into a list comprehension syntax.

I want a comprehension that would be equivalent to:

n = int(input())
l = []
for i in range(n):
    v1, v2 = input().split()
    l.append((int(v1), int(v2), i))

Here is what I've tried:

n = int(input())
l = [(int(v1), int(v2), i) for v1, v2 in input().split() for i in range(n)]

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)

Use tuple() and list comprehension:

num_tuples = int(input())
lst = [tuple([int(x) for x in input().split()] + [i]) for i in range(num_tuples)]
print(lst)

Example input:

2
1 2
3 4

Output:

[(1, 2, 0), (3, 4, 1)]

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...