如何解决TypeError: only size-1 arrays can be converted to Python scalars错误

错误根源 #

• math函数不兼容numpy数组：使用from math import *后，math库的sqrt、sin、cos仅能处理单个数值，无法直接作用于numpy数组（你的x是数组类型），这是触发类型错误的核心原因。
• f函数语法漏写运算符：np.exp(-2*x)后直接跟括号，缺少乘法运算符*，导致运算逻辑出错。
• 冗余参数无意义：f(x,y)定义里未使用y参数，调用时传入y纯属多余，还易造成混淆。

修正后的可运行代码 #

import numpy as np
import matplotlib.pyplot as plt

def f(x):
    return np.exp(-2*x) * ((np.sqrt(3)/6)*np.sin(2*np.sqrt(3)*x) + (1/2)*np.cos(2*np.sqrt(3)*x))

dx = 0.1
a = 0
b = 6
N = int((b - a)/dx)
x = np.linspace(a, b, N+1)
y = np.zeros(N+1)
y[0] = 6
y[1] = y[0]

for i in range(N-1):
    y[i+2] = (2 - 4*dx)*y[i+1] + (-1 + 4*dx - 16*dx**2)*y[i]

Y2 = f(x)
Err = abs(Y2 - y)

plt.plot(x, y, 'r', x, Y2, 'b')
plt.show()
plt.plot(x, Err)
plt.show()

重点修正说明 #

• 替换math函数为numpy版本：用np.sqrt、np.sin、np.cos替代math库的同名函数，它们天然支持numpy数组运算。
• 补全乘法运算符：在np.exp(-2*x)和后续表达式之间加上*，确保数组乘法逻辑正确。
• 删除冗余参数：把f(x,y)改成f(x)，调用时仅传入x即可。
• 规范matplotlib导入：改用import matplotlib.pyplot as plt，避免命名空间混乱，提升代码可读性。

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