Files
Mastering-Python-2e-code_2/CH_15_scientific_python/T_01_numba.rst
T
2022-05-05 18:25:55 +02:00

16 lines
289 B
ReStructuredText

>>> import numpy
>>> import numba
>>> numbers = numpy.arange(500, dtype=numpy.int64)
>>> @numba.vectorize([numba.int64(numba.int64)])
... def add_one(x):
... return x + 1
>>> numbers
array([ 0, 1, 2, ..., 498, 499])
>>> add_one(numbers)
array([ 1, 2, 3, ..., 499, 500])