python自定义函数输出数组