numpy扩展
numpy-ext的Python项目详细描述
NumPy扩展
NumPy的扩展库,它实现了NumPy中不存在的常见数组操作。在
npext.fill_na(...)
npext.drop_na(...)
npext.rolling(...)
npext.expanding(...)
npext.rolling_apply(...)
npext.expanding_apply(...)
# etc
文件
安装
常规安装:
pip install numpy_ext
为了发展:
^{pr2}$示例
下面是几个关于如何使用库的常见示例。其余的可以在documentation中找到。在
- 对提供的数组上的滚动窗口应用函数
importnumpyasnpimportnumpy_extasnpexta=np.array([0,1,2,3,4,5,6,7,8,9])window=3npext.rolling_apply(np.sum,window,a)>array([nan,nan,3.,6.,9.,12.,15.,18.,21.,24.])
- 与上面相同,但使用自定义函数、两个输入数组和使用
joblib
进行并行计算:
deffunc(array_first,array_second,param):return(np.min(array_first)+np.sum(array_second))*parama=np.array([0,1,2,3])b=np.array([3,2,1,0])npext.rolling_apply(func,2,a,b,n_jobs=2,param=-1)>array([nan,-5.,-4.,-3.])
- 与第一个示例相同,但使用rolling函数:
a=np.array([0,1,2,3,4,5,6,7,8,9])window=3rolls=npext.rolling(a,window,as_array=True)np.sum(rolls,axis=1)>array([nan,nan,3.,6.,9.,12.,15.,18.,21.,24.])
许可证
该软件是在麻省理工学院授权下发行的。在
- 项目
标签: