numpy数组而不是float

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