dfed7a844a
* implement nonzero for Boolean arrays * remove axtls from build script * extend nonzero to ndarrays of arbitrary dtype, and iterable, fix float tests * temporarily disable circuitpython tests * add nonzero documentation * Added test script for np.nonzero() Co-authored-by: Tejal Ashwini Barnwal <64950661+tejalbarnwal@users.noreply.github.com>
16 lines
No EOL
364 B
Python
16 lines
No EOL
364 B
Python
try:
|
|
from ulab import numpy as np
|
|
except:
|
|
import numpy as np
|
|
|
|
array = np.array(range(16)).reshape((4,4))
|
|
print(array)
|
|
print(array < 5)
|
|
print(np.nonzero(array < 5))
|
|
|
|
dtypes = (np.uint8, np.int8, np.uint16, np.int16, np.float)
|
|
|
|
for dtype in dtypes:
|
|
array = (np.arange(2, 12, 3, dtype=dtype)).reshape((2,2)) - 2
|
|
print(array)
|
|
print(np.nonzero(array)) |