Taking a list as input from user

Question

While taking a list as an input from the user, the input is SUPPOSED to be enclosed within square brackets as opposed to just typing the numbers This is my code for taking multiple inputs from the user for a list

a = list(map(int,input().strip().split(', ')))

The default input is [1, 2, 3, 4]. When I run my code, I get an error as:

ValueError: invalid literal for int() with base 10: '[1'

I know the reason it's happening is because '[1' is a an element that cannot be converted into int and it (the square bracket and 1) is being regarded as the first element of the list at index 0. Having the square bracket is absolutely essential for me, how can I resolve the error or just ignore the square brackets? Is there an ignore method or something similar that might work? Cheers!

score 2 · Answer 1 · answered Jun 02 '20 at 16:47

2

Strip off the brackets from the input:

a = [int(n) for n in input().strip('[]').split(',')]

answered Jun 02 '20 at 16:47

Terry Spotts

3,527
1
8
21

dzang · Answer 2 · 2020-06-02T16:56:03.607

0

You could remove it from the string before converting it to int:

a = list(map(int,
             input()
             .strip()
             .replace('[', '')
             .replace(']', '')
             .split(', ')
            )
        )

or you could also use a regular expression to get out all the numbers:

import re
a = list(map(int, re.findall(r'\d+', input())))

edited Jun 02 '20 at 16:56

answered Jun 02 '20 at 16:44

dzang

2,160
2
12
21

score 0 · Accepted Answer · answered Jun 02 '20 at 17:03

0

In a simple way you can use strip() to remove extra whitespace and '[ ]'.

a = list(map(int, input().strip(' []').split(',')))

answered Jun 02 '20 at 17:03

Nirav Solanki

16
2

Taking a list as input from user

3 Answers3