绘制复函数积分时出现TypeError错误的解决求助

错误原因 #

1. scipy.integrate.quad仅支持对输入输出均为标量的函数积分，你的代码中w(t)直接接收numpy数组t，且complex_function引用了全局数组t，导致被积函数返回数组，不符合quad的要求。
2. 直接传递数组t给w函数时，quad无法处理数组运算，触发only length-1 arrays can be converted to Python scalars错误。

修正方案 #

修改函数结构，让被积函数依赖局部参数而非全局变量，同时将标量处理的积分函数向量化以适配数组输入：

import matplotlib.pyplot as plt
import numpy as np
from scipy.integrate import quad

# 被积函数：接收标量f和t，以及其他参数，返回标量复数值
def complex_function(f, t_val, fs, a):
    exp_term = np.exp(2 * np.pi * 1j * f * t_val)
    tan_term = 1 / np.tan(f / fs - 1j * (a / (2 * np.pi * fs)))
    return exp_term * tan_term / (2 * np.pi * 1j * fs)

# 单个t值的积分计算：分开实部虚部处理复函数积分
def compute_w(t_val, fs, a, B):
    real_integral, _ = quad(lambda f: np.real(complex_function(f, t_val, fs, a)), -B/2, B/2)
    imag_integral, _ = quad(lambda f: np.imag(complex_function(f, t_val, fs, a)), -B/2, B/2)
    return real_integral + 1j * imag_integral

# 变量定义
B = 1
fs = 1
t = np.linspace(-25, 25, int(50 / fs))
a = 1

# 向量化积分函数，适配数组t
vectorized_w = np.vectorize(compute_w)
w_results = vectorized_w(t, fs, a, B)

# 绘制复函数的实部和虚部
plt.plot(t, np.real(w_results), 'b', label='实部')
plt.plot(t, np.imag(w_results), 'r', label='虚部')
plt.xlabel('t')
plt.ylabel('w(t)')
plt.legend()
plt.show()

关键修改点 #

• 移除全局变量：将t、fs、a作为参数传入complex_function，确保函数输入输出均为标量。
• 复函数积分处理：quad不直接支持复值函数，因此拆分实部和虚部分开积分后合并结果。
• 向量化适配：用np.vectorize将处理单个t值的函数转换为可处理数组的函数，也可改用for循环逐个计算数组元素。
• 绘图适配：matplotlib无法直接绘制复数值，需选择绘制实部、虚部或模长。

内容的提问来源于stack exchange，提问作者Kevs