monte_carlo_numpy_njit_parallel_pi.py

# Heitetään pisteitä satunnaisesti neliölle
# Osa pisteistä osuu neliön sisällä olevalle neljännesympyrälle
# Lasketaan tuloksista Piin arvo
# Juhani Kaukoranta 26.3.2019
import numpy as np
from numba import njit
import time
@njit(parallel = True)
def monte_carlo_pi(n):
    # neljännesympyrän sisällä pisteiden x- ja y-koordinaattien neliosumma < 1
    # pisteiden x- ja y-koordinaattien neliöiden summa < 1, tapausten lukumäärä
    accum = np.sum(np.random.random_sample(n)**2 + np.random.random_sample(n)**2 < 1)
    return 4.0*accum/n

pisteita = int(input("Kuinka monta pistettä arvotaan "))
time0 = time.perf_counter()
print("Pii = ",monte_carlo_pi(pisteita))
time1 = time.perf_counter()
print("Aikaa kului ",time1-time0," sekuntia")