python数组循环输入