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