python输入数组