我有4维风速数组（时间，ht，x，y），我试图以此为基础计算湍流动能。为此，在每个时间步长和每个高度处，我必须计算三个风分量的平均值，然后从每个网格点的速度中减去该平均值，计算每个网格点的TKE，然后在该点处求平均值高度。如下面的代码所示，这会导致很多循环。我运行了一次，花了一个星期的时间，不利于快速分析。
我的问题：是否可以矢量化这些循环中正在进行的进程，以帮助我的代码更快地运行？我见过关于1维和2维数组的示例，但是没有达到我尝试做的规模。
# calculating and plotting the correlation coeff's using a seaborn heatmap
corr = bos.corr()
sns.heatmap(corr, xticklabels=corr.columns, yticklabels=corr.columns, cmap='coolwarm')