Question

python输出最小和第二小的元素位置

我有一个列 a=[2,3,6,1,3,1,2]
最小的元素对应输出为2，第二小的元素输出为1，其他为0.
结果应该是：result=[1,0,0,2,0,2,1]
应为实际的列很大，所以想找快速、简单的方法
求帮忙啊~~~

Answer 1

a=[2,3,6,1,3,1,2]
b=sorted(list(set(a)))
for i,k in enumerate(a):
    if a[i]>b[1]:
        a[i]=0
    elif a[i]==b[1]:
        a[i]=1
    else:
        a[i]=2

Answer 2

a=[2,3,6,1,3,1,2]
b=list(set(a))           #将列表去重、排序，以便得到最小、第二小的值
dic={b[0]:2,b[1]:1}
result=[]
for data in a:
    if dic.has_key(data):
        result.append(dic[data])
    else:
        result.append(0)