如果输入为零,我想创建一个如下所示的数组:
[1,0,0,0,0,0,0,0,0,0]
如果输入为5:
[0,0,0,0,0,1,0,0,0,0]
对于上面我写道:
np.put(np.zeros(10),5,1)
但它不起作用.
有什么方法可以在一行中实现吗？