返回数组下标

'''
给定一个整数数组 nums和一个目标值 target，请你在该数组中找出和为目标值的那 
两个整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
target = nums1+nums2'''

def two_sum(nums,target):
    list = {}
    for i, j in enumerate(nums):
        # print("target - j的值是:{a}".format(a = target - j))
        if target - j in list:
            b = [list[target - j],i]
            # print("b的值：{b}".format(b=[list[target - j], i]))
            return b
        list[j] = i
        # print(list)
if __name__ == '__main__':
    sums=[1,2,3,4,9,6]
    target=9
    ts = two_sum(sums,target)
    print(ts)