AdvancedPythonProgramming/Chapter07/pi.py at master · Techlord-RCE/AdvancedPythonProgramming · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import multiprocessing
import random

samples = 10000000

def pi_serial():
    hits = 0

    for i in range(samples):
        x = random.uniform(-1.0, 1.0)
        y = random.uniform(-1.0, 1.0)

        if x**2 + y**2 <= 1:
            hits += 1

    pi = 4.0 * hits/samples
    return pi

def sample():
    x = random.uniform(-1.0, 1.0)
    y = random.uniform(-1.0, 1.0)

    if x**2 + y**2 <= 1:
        return 1
    else:
        return 0

def pi_apply_async():
    pool = multiprocessing.Pool()
    results_async = [pool.apply_async(sample)
                     for i in range(samples)]
    hits = sum(r.get() for r in results_async)

    pi = 4.0 * hits/samples
    return pi

def sample_multiple(samples_partial):
    return sum(sample() for i in range(samples_partial))

def pi_apply_async_chunked():
    ntasks = 10
    chunk_size = samples/ntasks
    pool = multiprocessing.Pool()
    results_async = [pool.apply_async(sample_multiple, [chunk_size])
                     for i in range(ntasks)]
    hits = sum(r.get() for r in results_async)

    pi = 4.0 * hits/samples
    return pi