给Python加入自己的函数

在日常研究中，我们有时候会写一些Python没有的，但是很多个脚本都需要用的函数，反复的复制函数太过麻烦，我们可以进行一些简单的操作来变成一个可以直接import的函数

1. 首先我们新建一个.py文件，把我们的函数放进去，比如我新建了一个fit_gaussian.py文件，里面是这样子

from astropy.modeling import models, fitting
import numpy as npdef fit_gaussian(data, bins):"""Function to employ gaussian fitting to input datainput: data(array or series in pandas)bins(number or array)output:g.mean.value(miu of fitting result)g.stddev.value(sigma of fitting result)g(fitting result. It's type is function. Tyr g(np.linspace(0,1,10)))"""hx, xedge = np.histogram(data,bins)xedge = (xedge[1:]+xedge[:-1])/2g_init = models.Gaussian1D(amplitude=np.max(hx), mean=np.mean(data), stddev=np.std(data))fit_g = fitting.LevMarLSQFitter()g = fit_g(g_init, xedge, hx)return g.mean.value, g.stddev.value, g

2. 接下来从终端cd 进入到该文件所在的路径，为了方便管理，最好新建一个文件夹来存放你的函数，然后每个函数都新建一个文件夹，把.py文件放进去，比如我是

cd /Users/xushuai/OneDrive/xu_function/fit_gaussian/

3. 接下来在该路径下新建一个setup.py文件，里面写

from setuptools import setupsetup(name="fit_gaussian",version="0.1",py_modules=["fit_gaussian"], 
)

这里的name和py_modules直接使用刚才新建的文件名即可

4. 在命令行里执行 

pip3 install --user .

 这样子就可以直接安装上去了，接下来在Python里直接

import fit_gaussianhelp(fit_gaussian)help(fit_gaussian.fit_gaussian)

就可以看到这个函数了，你也可以把一些功能相近的函数放到一个.py文件里