Python定义数组函数