python定义数组并输入