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对应数据,即拿到当周的数据,并凭此计算当周的最大值和最小值。

你可能感兴趣的:(10、apply_along_axis函数)