Python双变量求最大值报only size-1 arrays转scalars TypeError求解

错误原因 #

• 你使用了Python标准库math模块的sqrt函数，该函数仅支持输入单个数值标量，而你在计算时传入了整个长度为50的numpy数组d，无法直接转换为标量，因此触发该报错。
• 内层循环已经遍历d的每个元素赋值给dis，但计算x时仍然用了整个d数组，导致x每次都是长度为50的数组，既不符合遍历计算的逻辑，也加剧了类型不匹配问题。

修复方案 #

方案1：修正循环逻辑 #

如果需要遍历每个角度、每个距离计算对应的B值，调整代码如下：

import numpy as np
import matplotlib.pyplot as plt
from math import pi, sqrt

i = .5
l = .01
u = 4*pi*10**-7

angle = np.linspace(0, pi/2, 20)
d = np.linspace(0, .5, 50)

B = []

for ang in angle:
    for dis in d:
        # 把d替换为当前循环的单个距离值dis，math.sqrt可正常处理标量输入
        x = (u * i * np.cos(ang)) / (pi * sqrt((l/2)**2 + dis**2))
        B.append(x)

如果需要计算每个角度下所有距离对应的x的最大值，调整为：

B = []
for ang in angle:
    # 用numpy的sqrt处理数组输入，一次性计算当前角度下所有距离的x值
    x = (u * i * np.cos(ang)) / (pi * np.sqrt((l/2)**2 + d**2))
    B.append(x.max())

方案2：向量化计算（效率更高，无需嵌套循环） #

利用numpy广播机制直接计算所有角度和距离组合的x值，再按需取最大值：

import numpy as np
from math import pi

i = .5
l = .01
u = 4*pi*10**-7

angle = np.linspace(0, pi/2, 20)
d = np.linspace(0, .5, 50)

# 扩展维度适配广播规则，一次性计算所有(ang, dis)组合的x值
cos_angle = np.cos(angle)[:, np.newaxis]
d_sq = d[np.newaxis, :] **2
x = (u * i * cos_angle) / (pi * np.sqrt((l/2)**2 + d_sq))

# 取所有组合的全局最大值
global_max = x.max()
# 按角度维度取每个角度下的最大值
angle_wise_max = x.max(axis=1)

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