返回數組下標

'''
給定一個整數數組 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)