np.quantile 在 3d 阵列上运行非常缓慢

问题描述

我正在尝试使用 NumPy 的分位数函数在 3D NumPy 数组的第 3 维上计算 .95 分位数。我发现这需要很长时间来计算,并且在我的分析中造成了主要瓶颈。以下代码似乎重现了我正在经历的冗长计算:

import numpy as np

arr = np.random.normal(1,3,size=(2000,2000,200))

quants_95 = np.quantile(arr,q = .95,axis = 2)

这似乎比我的其他分析慢得多的原因之一是 np.quantile 似乎在单个内核上运行,而其他 numpy 函数(例如 np.correff)认运行在多个内核上。有谁知道在多核上运行 np.quantile 的方法,或者类似的功能

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)