CODEWITHSUNDEEP
pythonlistinsert
user1709173
user1709173

Reputation:

Inserting an item after each item in a list (Python)

Say I have a list like this:

a = [[1, 2, 3], [4, 5, 6]]
filler = ['cat']

How could I get 

b = [[1 ,2 ,3], ['cat'], [4, 5, 6], ['cat']]

As an output?

Upvotes: 1

Views: 418

Answers (9)

senderle
senderle

Reputation: 150957

I like the itertools-based solutions posted here, but here's an approach that doesn't require list comprehensions or itertools, and I bet it's super fast. 

new_list = [filler] * (len(a) * 2)
new_list[0::2] = a

Upvotes: 3

raton
raton

Reputation: 428

    python 3.2

    a = [[1, 2, 3], [4, 5, 6]]
    b = ['cat']
    _=[a.insert(x,b) for x in range(1,len(a)*2,2)]

Upvotes: -1

Óscar López
Óscar López

Reputation: 235994

Try this, as a one-liner:

from operator import add
a = [[1, 2, 3], [4, 5, 6]]
filler = ['cat']

reduce(add, ([x, filler] for x in a))
> [[1, 2, 3], ['cat'], [4, 5, 6], ['cat']]

Or even simpler, without using reduce:

sum(([x, filler] for x in a), [])
> [[1, 2, 3], ['cat'], [4, 5, 6], ['cat']]

Both solutions do the same: first, create a generator of [element, filler] and then flatten the resulting stream of pairs. For efficiency, the first step is performed using generators to avoid unnecessary intermediate lists.

UPDATE:

This post is a textbook example of why a troll must not be fed in an online forum. See the comments to see what I mean. It's better to just ignore the troll.

Upvotes: 0

Ashwini Chaudhary
Ashwini Chaudhary

Reputation: 250891

something like this using itertools.islice() and itertools.cycle():

cycle() is used to repeat an item, and used islice() cut the number of repeatation to len(a), and then use izip() or simple zip() over a and the iterator returned by islice() , this will return list of tuples.

you can then flatten this using itertools.chain().

In [72]: a
Out[72]: [[1, 2, 3], [4, 5, 6]]

In [73]: b
Out[73]: ['cat']

In [74]: cyc=islice(cycle(b),len(a))

In [75]: lis=[]

In [76]: for x in a:
    lis.append(x)
    lis.append([next(cyc)])
   ....:     

In [77]: lis
Out[77]: [[1, 2, 3], ['cat'], [4, 5, 6], ['cat']]

or:

In [108]: a
Out[108]: [[1, 2, 3], [4, 5, 6]]

In [109]: b
Out[109]: ['cat']

In [110]: cyc=islice(cycle(b),len(a))

In [111]: list(chain(*[(x,[y]) for x,y in izip(a,cyc)]))
Out[111]: [[1, 2, 3], ['cat'], [4, 5, 6], ['cat']]

Upvotes: 1

stranac
stranac

Reputation: 28226

I prefer to use itertools for stuff like this:

>>> import itertools as it
>>> a = [[1, 2, 3], [4, 5, 6]]
>>> filler = ['cat']
>>> list(it.chain.from_iterable(it.izip(a, it.repeat(filler))))
[[1, 2, 3], ['cat'], [4, 5, 6], ['cat']]

Upvotes: 3

Oleksandr Lysenko
Oleksandr Lysenko

Reputation: 304

result = [si for i in zip(a, [filler]*len(a)) for si in i]

Upvotes: 0

toxotes
toxotes

Reputation: 1464

a = [[1, 2, 3], [4, 5, 6]]
filler = ['cat']
out = []

for i in a:
    out.append(i)
    out.append(filler)

Upvotes: 0

arshajii
arshajii

Reputation: 129497

Here's an idea:

import itertools

a = [[1, 2, 3], [4, 5, 6]]
filler = ['cat']
print list(itertools.chain.from_iterable(zip(a, [filler] * len(a))))

Output:

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

Upvotes: 2

cwoebker
cwoebker

Reputation: 3288

Not pythonic but seems to work.

list = [[1, 2, 3], [4, 5, 6]]
result = []
for e in list:
    result.append(e)
    result.append(['cat'])
result.pop()

Found at this post: Add an item between each item already in the list

Upvotes: 2

Related Questions