micropython-ulab/snippets/tests/numpy/core/fromnumeric.py
2022-01-12 16:53:48 -08:00

27 lines
495 B
Python

import math
import sys
sys.path.append('.')
from snippets import numpy
from ulab import numpy as np
a = np.array([[1,2,3],[4,5,6]])
print(numpy.size(a))
print(numpy.size(a,1))
print(numpy.size(a,0))
print(numpy.prod([1, 10, 100, 5]))
print(numpy.prod([]))
print(numpy.prod([1.,2.]))
a = np.array([1,2,3])
print(numpy.nonzero(a))
b = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(numpy.nonzero(b > 3))
c = np.array([0,1,0,-1])
print(numpy.nonzero(c > 0))
print(numpy.nonzero(c < 0))