python自定义函数画图