python 处理数组输入