定时并行化数值计算

Question

我有一个简单的pi近似脚本，如下所示：

import numpy as np
import matplotlib.pyplot as plt
import time

start = 10
stop = 1000000
step = 100
exactsolution = np.pi


def montecarlopi(N=1000000):
    random_x = np.random.random(size = N)
    random_y = np.random.random(size = N)
    bod = np.array([random_x, random_y]).T
    square_area = N
    quarter_circle_area = np.count_nonzero(np.linalg.norm(bod, axis = 1)<=1)
    pi_approx = 4*quarter_circle_area/square_area
    return pi_approx


if __name__ == '__main__':
    times = []
    results = []
    attemps = np.arange(start = start, stop = stop, step = step)
    for i in attemps:
        start_time = time.time()
        results.append(montecarlopi(i))
        times.append(time.time()-start_time)

    absolute_errors = np.abs(np.array(results)-exactsolution)

我想知道根据我使用的随机尝试次数计算需要多长时间。正如您所看到的，我使用for循环来获得所需的每个计算时间，但这违背了Numpy的目的，大大减慢了我的代码。实际上，我只想在整个attemps数组上调用montecarlopi()，但这样就没有计算时间了。

有没有办法计算每个并行计算numpy所做的时间？

Answer 1

我使用了这里提供的答案中的计时代码：

• https://codereview.stackexchange.com/questions/165245/plot-timings-for-a-range-of-inputs

我只需在代码行中将labels更改为codecs：

empty_multi_index = pd.MultiIndex(levels=[[], []], codes=[[], []], names=['func', 'result'])

时序线性

然后，您可以使用以下命令运行整个计时实验

timings.plot_times([montecarlopi], inputs=np.arange(10, 1000000, 1000), repeats=3)

并得到如下的输出

​

​

计时日志空间

或者更清晰地使用logspacing

timings.plot_times([montecarlopi], inputs=np.logspace(1, 8, 8, dtype=np.int), repeats=3)

​

​