Skip to content
Advertisement

How to take in both input split by space(s) and by line(s)

for example I have this 2 inputs:

2 5 2

and:

2
5
2

How do I take all of them using the same code?

Advertisement

Answer

Read lines of input and split them until you get 3 total values.

inputs = []
while len(inputs) < 3:
    values = input().split()
    inputs.extend(values)

print(inputs) # this will print [2, 5, 2]
User contributions licensed under: CC BY-SA
10 People found this is helpful
Advertisement