18 lines
383 B
Python
18 lines
383 B
Python
try:
|
|
from ulab import numpy as np
|
|
except:
|
|
import numpy as np
|
|
|
|
dtypes = (np.uint8, np.int8, np.uint16, np.int16, np.float)
|
|
|
|
for dtype in dtypes:
|
|
print()
|
|
a = np.array([], dtype=dtype)
|
|
print(np.sort(a, axis=0))
|
|
print(list(np.argsort(a, axis=0)))
|
|
|
|
a = np.array([4, 1, 3, 2], dtype=dtype)
|
|
print(np.sort(a, axis=0))
|
|
print(list(np.argsort(a, axis=0)))
|
|
|
|
|