crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update openmc/stats/univariate.py

Browse source 
Co-authored-by: Paul Romano <paul.k.romano@gmail.com>
This commit is contained in:
Patrick Shriwise 2022-06-19 09:31:38 -05:00 • committed by GitHub
parent aba235e0b4
commit 0c3d30e6d1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

9
openmc/stats/univariate.py
View file

@ -139,14 +139,7 @@ class Discrete(Univariate):
def sample(self, n_samples=1, seed=None):
np.random.seed(seed)
if len(self.x) == 1:
return np.full((n_samples), self.x[0])
xi = np.random.rand(n_samples)
out = np.zeros_like(xi)
for i, c in enumerate(self.cdf()[:-1]):
out[xi >= c] = self.x[i]
return out
return np.random.choice(self.x, n_samples, p=self.p)
def normalize(self):
norm = sum(self.p)