python - Convert Single integer into a list

Question

Say I have the following lists:

a = 1
b = [2,3]
c = [4,5,6]

I want to concatenate them such that I get the following:

[1,2,3,4,5,6]

I tried the usual + operator:

>>> a+b+c
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'int' and 'list'

This is because of the a term. It is just an integer. So I convert everything to a list:

>>> [a]+[b]+[c]
[1, [2, 3], [4, 5, 6]]

Not quite what I'm looking for.

I also tried all the options in this answer, but I get the same int error mentioned above.

>>> l = [a]+[b]+[c]
>>> flat_list = [item for sublist in l for item in sublist]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
TypeError: 'int' object is not iterable

It should be simple enough, but nothing works with that term a. Is there any way to do this efficiently? It doens't necessarily have to be pythonic.

Question: why do you have inconsistent data types in the first place? Why not use lists consistently for all the variables? — Barmar
– Barmar, Commented Oct 1, 2017 at 21:44
My problem wasn't this simple. It was a part of a function where I updated the values of a list constantly (list of lists with variable lengths). It would be messy had I written it all here. But this was by basic error. — Ajay H
– Ajay H, Commented Oct 1, 2017 at 21:49

Barmar · Accepted Answer · 2017-10-01 21:36:58Z

14

There's nothing that will automatically treat an int as if it's a list of one int. You need to check whether the value is a list or not:

(a if type(a) is list else [a]) + (b if type(b) is list else [b]) + (c if type(c) is list else [c])

If you have to do this often you might want to write a function:

def as_list(x):
    if type(x) is list:
        return x
    else:
        return [x]

Then you can write:

as_list(a) + as_list(b) + as_list(c)

answered Oct 1, 2017 at 21:36

Barmar

789k57 gold badges555 silver badges669 bronze badges

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

2 Comments

Ajay H Over a year ago

I was actually running a+b+c for lists that change sizes. I only considered a. But at some point, even c became 1 integer. Got that from your function. Thanks

Barmar Over a year ago

Are you going to fix the question to have the correct value of a?

Ajax1234 · Accepted Answer · 2017-10-01 22:01:14Z

3

You can use itertools:

from itertools import chain

a = 1
b = [2,3]
c = [4,5,6]
final_list = list(chain.from_iterable([[a], b, c]))

Output:

[1, 2, 3, 4, 5, 6]

However, if you do not know the contents of a, b, and c ahead of time, you can try this:

new_list = [[i] if not isinstance(i, list) else i for i in [a, b, c]]
final_list = list(chain.from_iterable(new_list))

edited Oct 1, 2017 at 22:01

answered Oct 1, 2017 at 21:40

Ajax1234

71.7k9 gold badges67 silver badges110 bronze badges

1 Comment

Barmar Over a year ago

Will this work with a = 1, which is apparently his real situation?

Subham · Accepted Answer · 2021-02-27 07:54:04Z

1

Accepted answer is the best way. Adding another variation. Explanation in comments as well.

from collections.abc import Iterable


a = "fun" 
b = [2,3]
c = [4,5,6]


def flatten(lst):
    for item in lst:
        if isinstance(item,Iterable) and not isinstance(item,str): # checking if Iterable and taking care of strings as well
            yield from flatten(item)
        else:
            yield item

# test case:

res = []
res.extend([a]+[b]+[c]) # casting them into lists,  [a]+[b]+[c] [1, [2, 3], [4, 5, 6]]
print(list(flatten(res)))

Produces

['fun', 2, 3, 4, 5, 6]

[Program finished]

answered Feb 27, 2021 at 7:54

Subham

4111 gold badge6 silver badges14 bronze badges

Collectives™ on Stack Overflow

python - Convert Single integer into a list

3 Answers 3

2 Comments

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

2 Comments

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related