Split a list of tuples into sub-lists of the same tuple field [duplicate]

by

Use itertools.groupby:

import itertools
import operator

data=[(1, 'A', 'foo'),
    (2, 'A', 'bar'),
    (100, 'A', 'foo-bar'),

    ('xx', 'B', 'foobar'),
    ('yy', 'B', 'foo'),

    (1000, 'C', 'py'),
    (200, 'C', 'foo'),
    ]

for key,group in itertools.groupby(data,operator.itemgetter(1)):
    print(list(group))

yields

[(1, 'A', 'foo'), (2, 'A', 'bar'), (100, 'A', 'foo-bar')]
[('xx', 'B', 'foobar'), ('yy', 'B', 'foo')]
[(1000, 'C', 'py'), (200, 'C', 'foo')]

Or, to create one list with each group as a sublist, you could use a list comprehension:

[list(group) for key,group in itertools.groupby(data,operator.itemgetter(1))]

The second argument to itertools.groupby is a function which itertools.groupby applies to each item in data (the first argument). It is expected to return a key. itertools.groupby then groups together all contiguous items with the same key.

operator.itemgetter(1) picks off the second item in a sequence.

For example, if

row=(1, 'A', 'foo')

then

operator.itemgetter(1)(row)

equals 'A'.

As @eryksun points out in the comments, if the categories of the tuples appear in some random order, then you must sort data first before applying itertools.groupby. This is because itertools.groupy only collects contiguous items with the same key into groups.

To sort the tuples by category, use:

data2=sorted(data,key=operator.itemgetter(1))

More Related Contents:

  1. Why does “return list.sort()” return None, not the list?
  2. How to sort a list/tuple of lists/tuples by the element at a given index?
  3. How to sort a list of objects based on an attribute of the objects?
  4. What is the difference between `sorted(list)` vs `list.sort()`?
  5. Pythonic way to check if a list is sorted or not
  6. How to sort two lists (which reference each other) in the exact same way
  7. Combining two sorted lists in Python
  8. Why does ‘.sort()’ cause the list to be ‘None’ in Python? [duplicate]
  9. How to sort a list of tuples according to another list
  10. Custom Python list sorting
  11. case-insensitive list sorting, without lowercasing the result?
  12. Accessing the list while being sorted
  13. Does python have a sorted list?
  14. Sort a list of lists with a custom compare function
  15. Efficient method to calculate the rank vector of a list in Python
  16. How to group elements in python by n elements [duplicate]
  17. Identify groups of consecutive numbers in a list
  18. Group list by values [duplicate]
  19. How do I sort a list of objects based on an attribute of the objects?
  20. Given parallel lists, how can I sort one while permuting (rearranging) the other in the same way?
  21. Sort a list with a custom order in Python
  22. How to sort a list of strings in reverse order without using reverse=True parameter?
  23. Python Sort One List According to Another List
  24. Sort list while pushing None values to the end
  25. Retrieve the two highest item from a list containing 100,000 integers
  26. Creating sublists [duplicate]
  27. Nested lambda statements when sorting lists
  28. How to find the max number(s) in a list with tied numbers
  29. How can I sort a list, according to where its elements appear in another list?
  30. How to sort a list of dictionaries by a value of the dictionary in Python?

Leave a Comment