Transpose/Unzip Function (inverse of zip)?

python list matrix transpose
Cristian picture Cristian · Aug 21, 2008 · Viewed 167.7k times · Source

I have a list of 2-item tuples and I'd like to convert them to 2 lists where the first contains the first item in each tuple and the second list holds the second item.

For example:

original = [('a', 1), ('b', 2), ('c', 3), ('d', 4)]
# and I want to become...
result = (['a', 'b', 'c', 'd'], [1, 2, 3, 4])

Is there a builtin function that does that?

Answer

Patrick picture Patrick · Aug 21, 2008

zip is its own inverse! Provided you use the special * operator.

>>> zip(*[('a', 1), ('b', 2), ('c', 3), ('d', 4)])
[('a', 'b', 'c', 'd'), (1, 2, 3, 4)]

The way this works is by calling zip with the arguments:

zip(('a', 1), ('b', 2), ('c', 3), ('d', 4))

… except the arguments are passed to zip directly (after being converted to a tuple), so there's no need to worry about the number of arguments getting too big.

Related questions

Sum of list of lists; returns sum list

Let data = [[3,7,2],[1,4,5],[9,8,7]] Let's say I want to sum the elements for the indices of each list in the list, like adding numbers in a matrix column to get a single list. I am assuming that all lists in data are …

Read More
Co-occurrence matrix from nested list of words

I have a list of names like: names = ['A', 'B', 'C', 'D'] and a list of documents, that in each documents some of these names are mentioned. document =[['A', 'B'], ['C', 'B', 'K'],['A', 'B', 'C', 'D', 'Z']] I would …

Read More
How to create nested lists in python?

I know you can create easily nested lists in python like this: [[1,2],[3,4]] But how to create a 3x3x3 matrix of zeroes? [[[0] * 3 for i in range(0, 3)] for j in range (0,3)] or [[[0]*3]*3]*3 Doesn't seem right. There is no way to create …

Read More