Python group by

Do it in 2 steps. First, create a dictionary.

>>> input = [('11013331', 'KAT'), ('9085267', 'NOT'), ('5238761', 'ETH'), ('5349618', 'ETH'), ('11788544', 'NOT'), ('962142', 'ETH'), ('7795297', 'ETH'), ('7341464', 'ETH'), ('9843236', 'KAT'), ('5594916', 'ETH'), ('1550003', 'ETH')]
>>> from collections import defaultdict
>>> res = defaultdict(list)
>>> for v, k in input: res[k].append(v)
...

Then, convert that dictionary into the expected format.

>>> [{'type':k, 'items':v} for k,v in res.items()]
[{'items': ['9085267', '11788544'], 'type': 'NOT'}, {'items': ['5238761', '5349618', '962142', '7795297', '7341464', '5594916', '1550003'], 'type': 'ETH'}, {'items': ['11013331', '9843236'], 'type': 'KAT'}]

It is also possible with itertools.groupby but it requires the input to be sorted first.

>>> sorted_input = sorted(input, key=itemgetter(1))
>>> groups = groupby(sorted_input, key=itemgetter(1))
>>> [{'type':k, 'items':[x[0] for x in v]} for k, v in groups]
[{'items': ['5238761', '5349618', '962142', '7795297', '7341464', '5594916', '1550003'], 'type': 'ETH'}, {'items': ['11013331', '9843236'], 'type': 'KAT'}, {'items': ['9085267', '11788544'], 'type': 'NOT'}]

Note both of these do not respect the original order of the keys. You need an OrderedDict if you need to keep the order.

>>> from collections import OrderedDict
>>> res = OrderedDict()
>>> for v, k in input:
...   if k in res: res[k].append(v)
...   else: res[k] = [v]
... 
>>> [{'type':k, 'items':v} for k,v in res.items()]
[{'items': ['11013331', '9843236'], 'type': 'KAT'}, {'items': ['9085267', '11788544'],

Related Posts:

  1. How to groupby based on two columns in pandas?
  2. Pandas group-by and sum
  3. Count unique values per groups with Pandas
  4. Pandas group-by and sum
  5. Pandas ‘count(distinct)’ equivalent
  6. Count unique values using pandas groupby
  7. Count unique values per groups with Pandas [duplicate]
  8. pandas groupby sort within groups
  9. can’t multiply sequence by non-int of type ‘float’
  10. TypeError: ‘int’ object is not subscriptable
  11. Why there is no do while loop in python
  12. ImportError: DLL load failed: The specified module could not be found
  13. What is the difference between rw+ and r+
  14. ImportError: No module named requests
  15. Python: finding an element in a list [duplicate]
  16. Create a Pandas Dataframe by appending one row at a time
  17. Is it possible to break a long line to multiple lines in Python? [duplicate]
  18. What is the naming convention in Python for variable and function names?
  19. Convert bytes to a string
  20. Importing modules from parent folder
  21. Should I use np.absolute or np.abs?
  22. Writing a list to a file with Python
  23. How to run Conda?
  24. How do I check whether a file exists without exceptions?
  25. How can I get dictionary key as variable directly in Python (not by searching from value)?
  26. pygame.error: video system not initialized
  27. Python – Reading and writing csv files with utf-8 encoding
  28. Python’s equivalent of && (logical-and) in an if-statement
  29. Replace values in list using Python
  30. Trouble updating to Anaconda Navigator 1.10.0 (MacOS)
  31. Binary numbers in Python
  32. Django: ImproperlyConfigured: The SECRET_KEY setting must not be empty
  33. builtins.TypeError: must be str, not bytes
  34. How do you read from stdin?
  35. takes 1 positional argument but 2 were given
  36. What is dtype(‘O’), in pandas?
  37. Call a function from another file?
  38. What’s the difference between %s and %d in Python string formatting?
  39. What is the difference between Jupyter Notebook and JupyterLab?
  40. Python 3 ImportError: No module named ‘ConfigParser’
  41. keyerror 1 in my code
  42. ‘NoneType’ object has no attribute ‘group’
  43. How to fix ‘ValueError: list.remove(x): x not in list’ error in Python
  44. Group by index + column in pandas
  45. Why am I getting a FileNotFoundError?
  46. Anaconda Navigator won’t launch (windows 10)
  47. What are all possible pos tags of NLTK?
  48. Can’t open Jupyter notebook with Anaconda
  49. How to change dataframe column names in pyspark?
  50. Pickle , read in data , unsupported pickle protocol: 3 python 2.7
  51. pygame clock.tick() vs framerate in game main loop
  52. Converting XML to JSON using Python?
  53. Django – no such table exception
  54. How to import a module given the full path?
  55. Why does “pip install” inside Python raise a SyntaxError?
  56. Removing numbers from string
  57. importing external “.txt” file in python
  58. How can I remove Nan from list Python/NumPy
  59. How to implement a binary search tree in Python?
  60. TypeError: cannot unpack non-iterable NoneType object
  61. Search and replace a line in a file in Python
  62. Can’t import my own modules in Python
  63. What are the most common Python docstring formats?
  64. How do I install a pip package globally instead of locally?
  65. Python Array Rotation
  66. standard_init_linux.go:178: exec user process caused “exec format error”
  67. How to initialize weights in PyTorch?
  68. Can’t find module cPickle using Python 3.5 and Anaconda
  69. How can I check if two segments intersect?
  70. “Can’t convert ‘float’ object to str implicitly”
  71. Why I get ‘list’ object has no attribute ‘items’?
  72. tar: Unrecognized archive format error when trying to unpack flower_photos.tgz, TF tutorials on OSX
  73. Repeating a function in Python
  74. How can I convert a .py to .exe for Python?
  75. How do I make a dictionary with multiple keys to one value?
  76. Directing print output to a .txt file
  77. Nonlinear regression with python – what’s a simple method to fit this data better?
  78. ImportError: No module named mysql.connector using Python2
  79. Python read in string from file and split it into values
  80. range() for floats
  81. import pandas_datareader gives ImportError: cannot import name ‘is_list_like’
  82. How does tuple comparison work in Python?
  83. AttributeError: ‘float’ object has no attribute ‘split’4
  84. How to add pandas data to an existing csv file?
  85. Can sklearn random forest directly handle categorical features?
  86. What are the differences between numpy arrays and matrices? Which one should I use?
  87. Use different Python version with virtualenv
  88. Print empty line?
  89. write multiple lines in a file in python
  90. JSON object must be str, bytes or bytearray, not dict
  91. How can I create an object and add attributes to it?
  92. In Python, how do I determine if an object is iterable?
  93. How to implement the ReLU function in Numpy
  94. Python initializing a list of lists
  95. How do I compile my Python 3 app to an .exe?
  96. How to use python 3.5.1 with a MySQL database
  97. AttributeError: ‘module’ object has no attribute ‘main’ for tf.app.run()
  98. Error in Reading a csv file in pandas[CParserError: Error tokenizing data. C error: Buffer overflow caught – possible malformed input file.]
  99. Can I put a tuple into an array in python?
  100. What does an ‘r’ represent before a string in python?

Leave a Comment