Szukaj…


Korzystanie z funkcji z niestandardowej biblioteki C.

Mamy bibliotekę C o nazwie my_random która generuje losowe liczby z niestandardowej dystrybucji. Zapewnia dwie funkcje, których chcemy użyć: set_seed(long seed) i rand() (i wiele innych nie potrzebujemy). Aby użyć ich w Cython, musimy

  1. zdefiniuj interfejs w pliku .pxd i
  2. wywołaj funkcję w pliku .pyx.

Kod

test_extern.pxd

# extern blocks define interfaces for Cython to C code
cdef extern from "my_random.h":
    double rand()
    void c_set_seed "set_seed" (long seed) # rename C version of set_seed to c_set_seed to avoid naming conflict

test_extern.pyx

def set_seed (long seed):
    """Pass the seed on to the c version of set_seed in my_random."""
    c_set_seed(seed)

cpdef get_successes (int x, double threshold):
    """Create a list with x results of rand <= threshold
    
    Use the custom rand function from my_random.
    """
    cdef:
        list successes = []
        int i
    for i in range(x):
        if rand() <= threshold:
            successes.append(True)
        else:
            successes.append(False)
    return successes


Modified text is an extract of the original Stack Overflow Documentation
Licencjonowany na podstawie CC BY-SA 3.0
Nie związany z Stack Overflow