python输入数字成数组