CODEWITHSUNDEEP
pythonlistsplitelement
shawnngtq
shawnngtq

Reputation: 707

Split lists within list

Input:

[['Tryndamere', 'Barbarian', 'Fighter'],['Caitlyn', 'Sheriff', 'Marksmen'],...['Veigar', 'Midget', 'Mage']]

Expected output:

['Barbarian']['Caitlyn']['Fighter']['Mage']['Marksmen']['Midget']['Sheriff']['Tryndamere']['Veigar']...

Question: How to separate the lists within a list till they become individual elements? Thanks guys

Upvotes: 1

Views: 2606

Answers (2)

thefourtheye
thefourtheye

Reputation: 239683

Let's say the data looks like this

>>> lists = [
...     ['Tryndamere', 'Barbarian', 'Fighter'],
...     ['Caitlyn', 'Sheriff', 'Marksmen'],
...     ['Veigar', 'Midget', 'Mage']
... ]

then, you can use list comprehension like this

>>> from pprint import pprint
>>> pprint([[item] for c_list in lists for item in c_list])
[['Tryndamere'],
 ['Barbarian'],
 ['Fighter'],
 ['Caitlyn'],
 ['Sheriff'],
 ['Marksmen'],
 ['Veigar'],
 ['Midget'],
 ['Mage']]

Or using itertools.chain

>>> from itertools import chain
>>> pprint([[item] for item in chain.from_iterable(lists)])
[['Tryndamere'],
 ['Barbarian'],
 ['Fighter'],
 ['Caitlyn'],
 ['Sheriff'],
 ['Marksmen'],
 ['Veigar'],
 ['Midget'],
 ['Mage']]

But if you are thinking about flattening those lists into a single list, then please check this. You can easily extend the chain version to do that, like this

>>> pprint(list(chain.from_iterable(lists)))
['Tryndamere',
 'Barbarian',
 'Fighter',
 'Caitlyn',
 'Sheriff',
 'Marksmen',
 'Veigar',
 'Midget',
 'Mage']

Or the list comprehension version,

>>> pprint([item for c_list in lists for item in c_list])
['Tryndamere',
 'Barbarian',
 'Fighter',
 'Caitlyn',
 'Sheriff',
 'Marksmen',
 'Veigar',
 'Midget',
 'Mage']

Upvotes: 2

anon582847382
anon582847382

Reputation: 20391

Just use a list comprehension that iterates over the sublist also. Then you can convert that string to a list of length 1 just by wrapping it in square brackets.

>>> lst = [['Tryndamere', 'Barbarian', 'Fighter'], 
           ['Caitlyn', 'Sheriff', 'Marksmen'], 
           ['Veigar', 'Midget', 'Mage']]
>>> [[item] for sub in lst for item in sub]
[['Tryndamere'], ['Barbarian'], ['Fighter'], ['Caitlyn'], ['Sheriff'], ['Marksmen'], ['Veigar'], ['Midget'], ['Mage']]

Upvotes: 0

Related Questions