I have a simple question, I just want to store an input of multiple lines into an array using python, note that the first line of the input is telling me how many lines there will be, so if the first line of the input is 4, the whole input will be of 5 lines in total.
Example:
input:
4
1
2
3
4
output:
[1, 2, 3, 4]
I tried using n = list(map(int, input())) however when I print the whole list it only stores the first line of the input, and I need all the values.
Thank you.
Use a list comprehension, calling the input() function for each iteration, like so:
l = [int(input()) for _ in range(int(input()))]
output for print(l), with the first input being 4:
[1, 2, 3, 4]
Updated answer based on comment discussion.
num = input()
li = []
for _ in range(num):
data = int(input())
li.append(data)
print(li)
4
6
3
9
4
[ 6 , 3 , 9 , 4 ]
I don't know if this solution is too basic
output = []
tot = int(input())
for i in range(tot):
num=int(input())
output.append(num)
print(output)
If I input 4\n5\n6\n7\n2\n, I get the output:
[5, 6, 7, 2]
So, basically the first input gets the number of following inputs, and it is used to calculate the range of a for-loop, in which, for every iteration, an input and a list append is performed.
Please note how every input is returned in string format and needs to be converted into an integer.
This should work fine, looping in specific range and appending into a new list the inputs.
new_list = []
number_of_loop = int(input())
for _ in range(number_of_loop):
new_list.append(int(input()))
Output of print(new_list), if the number_of_loop was 5:
[1, 1, 1, 1, 1]
input() reads whole single line. You can extend it using sys.stdin.
Now, In your case, each line contains a single integer so there can be two cases:
#python 3.x
n = int(input())
ls = []
for i in range(n):
ls.append(int(input())
or
#python 3.x
import sys
n = int(input())
ls = list(map(int,sys.stdin.read().strip().split()))
if your input is like this
5
1
2
3
4
5
then value of n and ls will be
n = 5
ls = [1, 2, 3, 4, 5]
#python 3.x
import sys
ls = list(map(int, sys.stdin.read().strip().split())) #reading whole input from stdin then splitting
if your input is like this
1
2
3
4
5
then value of ls will be
ls = [1, 2, 3, 4, 5]
description = []
while input() != "n":
line = input()
description.append(line)
input("Do you want to continue (y/n)")
input()will be called only once this way, you will need some kind of an explicit loop for this.input()reads a single line from the standard input, yes? So of course anything you do with a single line of input will only process that line of input. You need to come up with a step-by-step process that handles multiple lines of input. Yes? So, think it through logically. What is the first step in this process: you read the first line, and it tells you how many more lines to process. Yes? So implement that. Now that you have that number, do you see how to proceed with the rest of the input?