將超過 300000k 行與 scipy.integrate quad 集成非常慢

Question

我在具有 366036 行的數據框“df_append”上使用以下定義，df 具有列 Fn 和

def f(r):
    return (((3*Fn[i])/(2*math.pi*a[i]**2))*(1-((r**2)/(a[i]**2)))**(1/2))

res = []
for i in range(len(df_append)):
    res += [(1/(-a[i]-a[i]))*(quad(f, a[i], -a[i])[0])]

而且速度非常慢，大約 10 分鍾。 關於如何加快速度的任何想法？

Answer 1

看看你的 function

它將繪制高度為3Fn[i]/2*pi*a[i]**2和寬度為2*a[i]的半省略號。

因此，您可以通過以下方式替換整個集成

res = []
for i in range(len(df_append)):
    res += [(3*Fn[i])/(8*a[i]**2)]

這應該會顯着加快速度，並在不到 1 秒的時間內運行。

如果你的Fn和a可以廣播算術運算，例如 numpy arrays，你可以簡單地寫

res = (3*Fn[i])/(8*a[i]**2);

它的運行時間不到 0.1 秒。