summaryrefslogtreecommitdiff
path: root/numpy/random/examples/cython/extending_distributions.pyx
diff options
context:
space:
mode:
authormattip <matti.picus@gmail.com>2019-11-19 22:23:00 -0800
committermattip <matti.picus@gmail.com>2019-11-19 22:23:00 -0800
commit521ea22552297345680c426c0204b17b644e6bd0 (patch)
tree883cc4a161230acc71692c71ff5fc69a98ffed35 /numpy/random/examples/cython/extending_distributions.pyx
parent7d75171c4e6fbb2ed0aebda10908a6a0b01d0c7e (diff)
downloadnumpy-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.pyx63
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