micropython-ulab/snippets/numpy/lib/function_base.py

14 lines
422 B
Python

from ulab import numpy as np
from ..core.multiarray import (asarray)
from ..core.overrides import set_module
@set_module('numpy')
def append(arr, values, axis=None):
arr = asarray(arr)
values = asarray(values)
if axis is None:
if len(arr.shape) != 1:
arr = arr.flatten()
values = values.flatten()
axis = len(arr.shape)-1
return np.concatenate((arr, values), axis=axis)