Group by multiple keys and summarize/average values of a list of dictionaries

by

To get the aggregated results

from itertools import groupby
from operator import itemgetter

grouper = itemgetter("dept", "sku")
result = []
for key, grp in groupby(sorted(input_data, key = grouper), grouper):
    temp_dict = dict(zip(["dept", "sku"], key))
    temp_dict["qty"] = sum(item["qty"] for item in grp)
    result.append(temp_dict)

from pprint import pprint
pprint(result)

Output

[{'dept': '001', 'qty': 200, 'sku': 'bar'},
 {'dept': '001', 'qty': 400, 'sku': 'foo'},
 {'dept': '002', 'qty': 900, 'sku': 'baz'},
 {'dept': '002', 'qty': 600, 'sku': 'qux'},
 {'dept': '003', 'qty': 700, 'sku': 'foo'}]

And to get the averages, you can simply change the contents inside the for loop, like this

temp_dict = dict(zip(["dept", "sku"], key))
temp_list = [item["qty"] for item in grp]
temp_dict["avg"] = sum(temp_list) / len(temp_list)
result.append(temp_dict)

Output

[{'avg': 200, 'dept': '001', 'sku': 'bar'},
 {'avg': 200, 'dept': '001', 'sku': 'foo'},
 {'avg': 450, 'dept': '002', 'sku': 'baz'},
 {'avg': 600, 'dept': '002', 'sku': 'qux'},
 {'avg': 700, 'dept': '003', 'sku': 'foo'}]

Suggestion: Anyway, I would have added both the qty and avg in the same dict like this

temp_dict = dict(zip(["dept", "sku"], key))
temp_list = [item["qty"] for item in grp]
temp_dict["qty"] = sum(temp_list)
temp_dict["avg"] = temp_dict["qty"] / len(temp_list)
result.append(temp_dict)

Output

[{'avg': 200, 'dept': '001', 'qty': 200, 'sku': 'bar'},
 {'avg': 200, 'dept': '001', 'qty': 400, 'sku': 'foo'},
 {'avg': 450, 'dept': '002', 'qty': 900, 'sku': 'baz'},
 {'avg': 600, 'dept': '002', 'qty': 600, 'sku': 'qux'},
 {'avg': 700, 'dept': '003', 'qty': 700, 'sku': 'foo'}]

More Related Contents:

  1. how do I make two dictionary keys with value as list [closed]
  2. How to name a file using a value in a list in a dictionary?
  3. I don’t know how to display step 3 for this assignment
  4. How do I initialize a dictionary of empty lists in Python?
  5. Access a particular field in arbitrarily nested JSON data [duplicate]
  6. How to return dictionary keys as a list in Python?
  7. Dictionary: Get list of values for list of keys
  8. In Python, when to use a Dictionary, List or Set?
  9. Create a list with initial capacity in Python
  10. python filter list of dictionaries based on key value
  11. How can I access the nested data in this complex JSON, which includes another JSON document as one of the strings?
  12. List of tuples to dictionary [duplicate]
  13. How do I combine two lists into a dictionary in Python? [duplicate]
  14. Converting list of tuples into a dictionary
  15. [] and {} vs list() and dict(), which is better?
  16. Merging dictionary value lists in python
  17. Sorting a dictionary with lists as values, according to an element from the list
  18. How do I serialize a Python dictionary into a string, and then back to a dictionary?
  19. Convert list of strings to dictionary
  20. Using a dict to translate numbers to letters in python
  21. How can I make a dictionary (dict) from separate lists of keys and values?
  22. How can I avoid “RuntimeError: dictionary changed size during iteration” error?
  23. What are dict_keys, dict_items and dict_values?
  24. Creating a dictionary with list of lists in Python
  25. How do I send list of dictionary as Body parameter in FastAPI?
  26. Python – How to convert only numbers in a mixed list into float?
  27. How can I convert string values from a dictionary, into int/float datatypes?
  28. How do I send list of dictionary as Body parameter together with Files in FastAPI?
  29. How to convert only numbers in a mixed list into float
  30. In what case would I use a tuple as a dictionary key? [closed]

Leave a Comment