Read input as nested list in python

Question

I am new to python. I want to read the input from stdin as nested list.

Stdin:

student1 90
student2 85
student3 98

My list should be as follows:

student = [['student1',90],['student2',85],['student3',98]]

Is there any way I can read the input using list comprehension without needing any extra space.

jpp · Accepted Answer · 2018-02-16 11:53:51Z

1

This is one way.

mystr = 'student1 90\nstudent2 85\nstudent3 98'

[[i[0], int(i[1])] for i in (k.split() for k in mystr.split('\n'))]

# [['student1', 90], ['student2', 85], ['student3', 98]]

edited Feb 16, 2018 at 11:53

answered Feb 16, 2018 at 11:46

jpp

166k37 gold badges301 silver badges362 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

gonczor Over a year ago

You have to change second element in the list to integer. Nice solution beside this.

gonczor · Accepted Answer · 2018-02-16 11:48:43Z

0

my_list = []
while some_condition:
  read = input()
  my_list.append(read.split())
  my_list[-1][1] = int(my_list[-1][1])

Now let's break it down:

Create an empty list,
Keep on reading from stdin (you get str here)
add read element to your list by splitting it (it will create a new list).
Turn second element of last inserted item to integer.

EDIT That's what it runs like:

In [1]: my_list = []
   ...: while True:
   ...:   read = input()
   ...:   my_list.append(read.split())
   ...:   my_list[-1][1] = int(my_list[-1][1])
   ...:   print(my_list)
   ...:   
student 1
[['student', 1]]
student 2
[['student', 1], ['student', 2]]
student 3
[['student', 1], ['student', 2], ['student', 3]]

answered Feb 16, 2018 at 11:48

gonczor

4,1841 gold badge25 silver badges53 bronze badges

Collectives™ on Stack Overflow

Read input as nested list in python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related