функция с использованием NumPy

Question

Помогите решить задачу с курса. Мой вариант не работает. Функцию, которая будет принимать два аргумента:

два массива,
массив и список,
два списка.

В первом случае суммируйте их и верните результат.
Во втором случае верните строку с предупреждением One argument is a list.
В последнем случае верните Both arguments are lists, not arrays.

import numpy as np

def custom_sum(arg1, arg2):

    if type(arg1) == np.ndarray and type(arg2) == np.ndarray:
        return arg1 + arg2

Как необходимо продолжить?

Answer 1

import numpy as np

def custom_sum(arg1, arg2):

if type(arg1) == np.ndarray and type(arg2) == np.ndarray:
    result = arg1 + arg2
    return result
elif type(arg1) == np.ndarray and type(arg2) == list:
    return ('One argument is a list')
elif type(arg2) == np.ndarray and type(arg1) == list:
    return ('One argument is a list')
elif type(arg2) == list and type(arg1) == list:
    return ('Both arguments are lists, not arrays')

БЛОГ НА HUSL

функция с использованием NumPy

Ответы (1 шт):