I am getting input in the form of:
4 11111
I am using a,b = map(int,raw_input().split()) to store 4 in a and 11111 in b.
But I want b to be a list, how am I supposed to do that ?
3 Answers
You can just use list :
>>> list('11111')
['1', '1', '1', '1', '1']
But in that case you can not use map function because it just apply one function on its iterable argument and in your code it convert the whole of '11111' to integer so you have tow way :
- create
bas a list of string ones :
inp=raw_input().split() a,b = int(inp[0]),list(inp[1])
- if you want a list of integer 1's use
map:
>>> map(int,'11111') [1, 1, 1, 1, 1]
2 Comments
John Constantine
But I need the 1`s in list to be integers
Bhargav Rao
Always provide links to the docs. docs.python.org/2/library/functions.html#map
You can try this in two steps:
a, b = raw_input().split()
a, b = int(a), map(int, b)
print a
print b
Returns: 4 and [1, 1, 1, 1, 1]
Comments
a, b = map(int, raw_input().strip().split())
char list:
l = list(str(b))
int list:
l = [int(i) for i in str(b)]
5 Comments
Padraic Cunningham
there is no need to strip, you also end up with strings so why cast to int and then turn them back into strings again?
ratherlargeguy
fyi, ozgur added strip()
Padraic Cunningham
who added the
l = list(str(b))?ratherlargeguy
I wrote everything except strip().
Padraic Cunningham
a, b = "4 11111".split();a, b = int(a), list(b) will be more efficient.
bto be a list of what?[11111],[1, 1, 1, 1, 1]? And what input are you expecting in general?