leetcode-groupAnagrams.py

class Solution(object):
    def groupAnagrams(self, strs):
        """
        :type strs: List[str]
        :rtype: List[List[str]]
        """
        ans=[]
        dict1 = {}
        if strs==[""]:
            return [[""]]
        elif len(strs) == 1:
            return [strs]
        else :
            
            for word in strs:
                sorted_word = "".join(sorted(word))
                if sorted_word not in dict1:
                    dict1[sorted_word]=[word]
                else:
                    dict1[sorted_word].append(word)
        
            for words in dict1.values():
                ans.append(words)
        return ans

        dict1= {}
        for word in strs:
            sorted_word = "".join(sorted(word))
            if sorted_word not in dict1:
                dict1[sorted_word]=[word]
            else:
                dict1[sorted_word].append(word)
        result = []
        for words in dict1.values():
            result.append(words)
        return result