dialectstat написав:ping написав:якщо z це масив, то яким макаром Ви хочете його в другій функції порівнювати з числом?
я кожен елемент хочу порівнювати і кожномо елементу повернути певне значення, перша функція ж працює з таким самим масивом.
перша функція приймає на вхід масив і повертає масив. і, якщо Ви глянете її код, то там не буде порівння масиву з числом, а буде порівняння елементу масива з числом.
а Вам здаєьтся треба отаке:
https://docs.scipy.org/doc/numpy/refere … orize.html
class numpy.vectorize(pyfunc, otypes=None, doc=None, excluded=None, cache=False, signature=None)[source]¶
Generalized function class.
Define a vectorized function which takes a nested sequence of objects or numpy arrays as inputs and returns an single or tuple of numpy array as output. The vectorized function evaluates pyfunc over successive tuples of the input arrays like the python map function, except it uses the broadcasting rules of numpy.
The data type of the output of vectorized is determined by calling the function with the first element of the input. This can be avoided by specifying the otypes argument.
>>> def myfunc(a, b):
... "Return a-b if a>b, otherwise return a+b"
... if a > b:
... return a - b
... else:
... return a + b
>>> vfunc = np.vectorize(myfunc)
>>> vfunc([1, 2, 3, 4], 2)
array([3, 4, 1, 2])