diff options
author | mattip <matti.picus@gmail.com> | 2019-11-19 22:23:00 -0800 |
---|---|---|
committer | mattip <matti.picus@gmail.com> | 2019-11-19 22:23:00 -0800 |
commit | 521ea22552297345680c426c0204b17b644e6bd0 (patch) | |
tree | 883cc4a161230acc71692c71ff5fc69a98ffed35 /numpy/random/examples/cython/extending_distributions.pyx | |
parent | 7d75171c4e6fbb2ed0aebda10908a6a0b01d0c7e (diff) | |
download | numpy-521ea22552297345680c426c0204b17b644e6bd0.tar.gz |
MAINT: move numpy/random/examples -> numpy/random/_examples
Diffstat (limited to 'numpy/random/examples/cython/extending_distributions.pyx')
-rw-r--r-- | numpy/random/examples/cython/extending_distributions.pyx | 63 |
1 files changed, 0 insertions, 63 deletions
diff --git a/numpy/random/examples/cython/extending_distributions.pyx b/numpy/random/examples/cython/extending_distributions.pyx deleted file mode 100644 index d17da45c1..000000000 --- a/numpy/random/examples/cython/extending_distributions.pyx +++ /dev/null @@ -1,63 +0,0 @@ -#!/usr/bin/env python -#cython: language_level=3 -""" -This file shows how the to use a BitGenerator to create a distribution. -""" -import numpy as np -cimport numpy as np -cimport cython -from cpython.pycapsule cimport PyCapsule_IsValid, PyCapsule_GetPointer -from numpy.random._bit_generator cimport bitgen_t -from numpy.random import PCG64 - - -@cython.boundscheck(False) -@cython.wraparound(False) -def uniforms(Py_ssize_t n): - """ Create an array of `n` uniformly distributed doubles. - A 'real' distribution would want to process the values into - some non-uniform distribution - """ - cdef Py_ssize_t i - cdef bitgen_t *rng - cdef const char *capsule_name = "BitGenerator" - cdef double[::1] random_values - - x = PCG64() - capsule = x.capsule - # Optional check that the capsule if from a BitGenerator - if not PyCapsule_IsValid(capsule, capsule_name): - raise ValueError("Invalid pointer to anon_func_state") - # Cast the pointer - rng = <bitgen_t *> PyCapsule_GetPointer(capsule, capsule_name) - random_values = np.empty(n, dtype='float64') - with x.lock, nogil: - for i in range(n): - # Call the function - random_values[i] = rng.next_double(rng.state) - randoms = np.asarray(random_values) - - return randoms - -# cython example 2 - -@cython.boundscheck(False) -@cython.wraparound(False) -def uint16_uniforms(Py_ssize_t n): - cdef Py_ssize_t i - cdef bitgen_t *rng - cdef const char *capsule_name = "BitGenerator" - cdef double[::1] random_values - - x = PCG64() - capsule = x.capsule - if not PyCapsule_IsValid(capsule, capsule_name): - raise ValueError("Invalid pointer to anon_func_state") - rng = <bitgen_t *> PyCapsule_GetPointer(capsule, capsule_name) - random_values = np.empty(n, dtype='uint32') - # Best practice is to release GIL and acquire the lock - with x.lock, nogil: - for i in range(n): - random_values[i] = rng.next_uint32(rng.state) - randoms = np.asarray(random_values) - return randoms |