Numpy массив вместо поплавка

Question

Я написал простую функцию

def rect(t):
    field = np.zeros((len(t),3))
    for i in range(len(t)):
        if t[i] >=-10 and t[i] <=10:
            field[i,1] = 1
        else:
            field[i,1] = 0
    return field

и хочу интегрировать ее

from scipy import integrate

def S_elem(t):
    return rect(t)[:,1]*integr

integrate.quad(S_elem, -10, 10)

Я получаю эту ошибку

TypeError: object of type 'float' has no len()

Я знаю, что могуне использовать массив numpy, но он мне нужен для других целей.Как я могу выполнить интеграцию без удаления типа numpy массива?

Answer 1

Вы должны преобразовать ваш float в массив numpy:

def rect(t):
    t = np.atleast_1d(t)
    field = np.zeros((len(t), 3))
    field[:, 1] = abs(t) <= 10
    return field