random.normalで自然数の乱数を作りたい

Question

前提・実現したいこと

初心者なので、基本的なもので申し訳ありませんがご教授ください。
random.normalを使って２＊２の行列の自然数のみの乱数を作りたいのですが、random.normalのみでは乱数が小数になってしまいます。自然数のみにするにはどのようにしたらよいでしょうか？
よろしくお願いいたします

発生している問題・エラーメッセージ

自然数のみの乱数を生成する方法がわからない

該当のソースコード

import random
import numpy as np
pts=np.random.normal(325,400,(2,2))

補足情報（FW/ツールのバージョンなど）

os:mac
python:3.7 バージョン

Answer 1

numpy.random.randintなどを用いると良いのではないでしょうか。

関数自体は整数の一様分布を生成するものなので、[1, 101)のような区間を設定して使います。

import numpy as np
>>> np.random.randint(1, 101, size=(2, 2))
array([[82, 53],
       [ 5, 68]])

numpy.random.mtrand.RandomState.randint — NumPy v1.17 Manual

Comments

[TakaKan]

ご回答ありがとうございます！
書き漏れてしまっていて大変申し訳ないのですが、実は上にあげたコードの325のようにある値に乱数が偏るように作りたいのでrandom.normalを使いたいのです。
random.normalを使って自然数のみにする方法はないでしょうか
教えていただけると嬉しいです

[hayataka2049]

分布の厳密さは保証できませんが、整数に型変換すればいいのでは？
>>> import numpy as np
>>> pts = np.random.normal(325,400,(2,2)).astype(np.int64)
>>> pts
array([[113, 30],
[285, -16]])
でも、正規分布を使う限り「自然数のみ」はきついですね。

[TakaKan]

ありがとうございます！
整数のみにすることができました。型変換すればよいのですか。思いつきませんでした
やはり正の数のみにするのは厳しいんですね…

[hayataka2049]

分布を変えればなります。切断正規分布あたりが適当でしょうか。
numpyにはありませんがscipyにはあるので、こちらが使えるかも知れません。
https://docs.scipy.org/doc/scipy/reference/generated/scipy.stats.truncnorm.html

[R.Shigemori]

横から失礼します。

np.random.poissonなら自然数が出力されます。ただし、用いる分布はpoisson分布なので正規分布とは形状が異なりますが･･･

[TakaKan]

ありがとうございます。
参考にさせていただきます