在不同的列表中追加索引为某个整数索引的列表元素

Question

以下是我的错误代码：

   n = int(input())
   coins = list(map(int,input().split()))
   workertype = list(map(int,input().split()))
   a1 = []
   a2 = []
   a3 = []
   for i in range(n):
     if(workertype[i]==1):
       a1.append(coins[i])

   elif(workertype[i]==2):
      a2.append(coins[i])

   elif(workertype[i]==3):
      a3.append(coins[i])
   print(a1,a2,a3)

coins和workertype是两个长度相等的列表。我希望列表a1包含列表coins的那些元素，其索引是列表1中workertype（所有1＆＃39; s）的索引。 a2和a3相同，但适用于所有2和3。

这是我的输出

    1(#input for n)
    5 6 7 5 9 10 8(#coins)
    1 2 3 3 2 2 1(#workertype)
    [5] [] [](a1, a2, a3)

而不是这个我想要a1，a2，a3分别是[5,1]，[6,9,10]，[7,5]。

Answer 1

如果你输入7（硬币和workertype的长度）为n。