python输入一个数组