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