grouped_anagrams.py

from typing import Optional, List
from collections import defaultdict
from itertools import groupby

def grouped_anagram(anagram_arr: List[str]) -> List[List[str]]:
    w_group = defaultdict(list)
    for key_sum, word in groupby(anagram_arr, lambda w: sum(map(ord, w))):
        w_group[key_sum] += list(word)
    return list(w_group.values())

print(grouped_anagram(["eat", "tea", "tan", "ate", "nat", "bat"]))