Pythonで素数を生成 (2017/05/28)

しょうらいのゆめ: りーまんよそうをしょうめいする。

○以下のコマンドをIPythonコンソールに入力して実行する。

import numpy as np

def get_prime_number(n):
    natural_number = np.arange(n) + 1.0
    natural_number[0] = np.nan
    for i in range(1, int(np.floor(n/2))):
        natural_number[(natural_number!=natural_number[i])
        & (natural_number%natural_number[i]==0)] = np.nan
    prime_number = natural_number[~np.isnan(natural_number)]
    prime_number = prime_number.astype(int)
    return prime_number

prime_number = get_prime_number(1000)
print(prime_number)

[  2   3   5 ..., 983 991 997]