Question

利用python把数组中的每一个元素拓展为一个数组

Python新手小白一个，各位大神，请问如何使用python把数组中的每一个元素拓展为一个数组呢？具体来说，y=[0 2 1 2 0 1 0 0 2 1]，我现在想把0变为[1 0 0]，把1变为[0 1 0]，把2变为[0 0 1]，不管使用循环语句还是内嵌函数都可以，请各位大神帮着解决下，目前没有思路啊，谢谢！

Answer 1

if __name__ =='__main__':
   y = [0, 2, 1, 2, 0, 1, 0, 0, 2, 1]
   result =[]
   for i in range(len(y)):
      if y[i] == 0:
         result.extend([1,0,0])
      elif y[i] == 1:
         result.extend([0,1,0])
      elif y[i] == 2:
         result.extend([0,0,1])
      else:
         result.append(y[i])
   print(result)

这里新开了一个result,增加了内存开销

追问

这位大神，非常感谢您的帮助，可是按照您的代码实现的是[1,0,0,0,0,1,0,1,0,0,0,1,1,0,0,0,1,0,……]，可是我想实现的是[[1,0,0],[0,0,1],[0,1,0]……]，请问该怎样实现呢？非常感谢您的帮助！