python - 不明白为什么我会收到 'numpy.ndarray object not callable' 错误？

Question

我有一个代码，它使用 numpy 和 hermval 以及多个函数来计算 psi 最后给定参数。但我一直收到错误 numpy.ndarray object not callable，我真的不明白为什么会这样。以下是我的代码的相关部分:

import numpy as np
import math
from numpy.linalg import eigh
from numpy.polynomial.hermite import hermval

def matrices(N, lam):
    H_0 = np.zeros([N+1, N+1])
    x_four_matrix = np.zeros([N+1, N+1])
    for n in range(N+1):
        for m in range(N+1):
            if n == m:
                H_0[n][m] = n + 0.5
                x_four_matrix[n][m] = (6.0*n**2 + 6.0*n + 3.0)/4.0
            elif n == m-2:
                x_four_matrix[n][m] = np.sqrt((n+1)*(n+2))*(n+1.5)
            elif n == m+2:
                x_four_matrix[n][m] = (n-0.5)*np.sqrt(n*(n-1))
            elif n == m-4:
                x_four_matrix[n][m] = np.sqrt((n+1)*(n+2)*(n+3)*(n+4))/4.0
            elif n == m+4:
                x_four_matrix[n][m] = np.sqrt((n-3)*(n-2)*(n-1)*n)/4.0
    return H_0, x_four_matrix

def H_lam(N, lam):
    return matrices(N, lam)[0] + lam*matrices(N, lam)[1]

# Solve for eigenvalues (energies)
def lowest_eigenvals(N, n, lam):
    lowest_eigs = []
    eigenvals = eigh(H_lam(N, lam))[0]
    eigenvals.sort()
    for i in range(n):
        lowest_eigs.append(eigenvals[i])
    return lowest_eigs

# Solve for eigenvectors
def lowest_eigenvectors(N, n, lam):
    lowest_vecs = []
    for i in range(len(lowest_eigenvals(N, n, lam))):
        for j in range(len(eig(H_lam(N, lam))[0])):
            if lowest_eigenvals(N, n, lam)[i] == eigh(H_lam(N, lam))[0][j]:
                lowest_vecs.append(eigh(H_lam(N, lam))[1][j])
    return np.array(lowest_vecs)

def N_coeff(i):
    return 1.0/np.sqrt(2**i*math.factorial(i)*np.sqrt(np.pi))

# for E_0 (first eigenfunction):
def psi(x, lowest_eigenvectors, i):
    herm_coeffs = [element*N_coeff(i) for element in lowest_eigenvectors(N, n, lam)[i]]
    return np.exp((x**2)/2.0)*hermval(x, herm_coeffs)

print [element*N_coeff(0) for element in lowest_eigenvectors(100, 4, 0.1)[0]]
print psi(1.0, lowest_eigenvectors(100, 4, 1.0), 0) # for lambda = 1

然后在我最后的 print 语句中，我得到 TypeError: 'numpy.ndarray' object is not callable 来自 herm_coeffs我最后一个函数中的行。但我不确定为什么会这样，因为倒数第二个 print 语句打印正确!这是怎么回事？

这是回溯:

TypeError                                 Traceback (most recent call last)
<ipython-input-350-04692f269a26> in <module>()
     13 # print [element*N_coeff(0) for element in lowest_eigenvectors(100, 4, 0.1)[0]]
     14 
---> 15 print psi(1.0, lowest_eigenvectors(100, 4, 0.1), 0)

<ipython-input-350-04692f269a26> in psi(x, lowest_eigenvectors, i)
      7 # for E_0 (first eigenfunction):
      8 def psi(x, lowest_eigenvectors, i):
----> 9     herm_coeffs = [element*N_coeff(i) for element in lowest_eigenvectors(N, n, lam)[i]]
     10     return np.exp((x**2)/2.0)*hermval(x, herm_coeffs)
     11 

TypeError: 'numpy.ndarray' object is not callable

Answer 1

lowest_eigenvectors 函数 psi 中的参数名称与函数 lowest_eigenvectors 冲突。

编辑:自从 psi 以来，您似乎不需要将 psi 函数传递给 lowest_eigenvectors功能在同一lexical scope内作为 lowest_eigenvectors。