10、apply_along_axis函数

Numpy中的apply_along_axis函数,在数据分析中非常实用,它调用给定的自定义函数,让数组元素按特定轴进行计算:

np.apply_along_axis(自定义func, axis, arr, 自定义func中需要的其他参数)。

图1 拿到所需的数据集及对应索引
图2 apply_along_axis函数的应用

首先我们建立了一个summarize自定义函数,要求使用apply_along_axis函数在1轴上对week_index索引数组进行summarize的计算。详细说明:1⃣️按1轴计算,意思是我们的自定义函数作用是列;即,week_index[0]提取到的就是索引[1,6,11],也就是三个周的周一索引,并在o数组中找到对应的三个周一的开盘价格。2⃣️在apply_along_axis 1轴下,我们的week_index索引每行独立,即每周独立。我们通过take函数拿到了h数组中的week_index对应数据,即拿到当周的数据,并凭此计算当周的最大值和最小值。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容