Leetcode 1512. Number of Good Pairs [Python]

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:
        dic = collections.defaultdict(list)
        ans = 0
        def addition(n):
            res = 0
            while n != 0:
                res += n
                n -= 1
            return res
        for index, num in enumerate(nums):
            dic[num].append(index)
        for num,ls in dic.items():
            if len(ls) >= 2:
                ans += addition(len(ls)-1)
        return ans