如何在Python for循环中隔离函数计算以避免numpy广播形状冲突？

你的核心问题是numpy数组广播规则不匹配：当你传入长度为75的mu和intens数组时，一维的x（形状(4001,)）无法和这两个数组直接进行运算，导致形状冲突报错。下面提供两种解决方案，分别对应高效的向量化计算和你想要的"独立逐个计算"逻辑：

方案1：利用numpy向量化计算（推荐，效率更高） #

我们只需要调整x的形状，让它支持自动广播，就能一次性计算所有75个高斯峰，不需要循环：

import numpy as np

def gaussian(intens, mu):
    x = np.arange(4001).reshape(-1, 1)  # 将x转为(4001, 1)的二维数组
    sig = 50
    return intens * np.exp(-np.power(x - mu, 2.) / (2 * np.power(sig, 2.)))

# 定义你的75组峰值位置和强度
mu = np.array([106.2516, 169.2317, 179.4433, 210.1843, 225.1875, 237.6963, 261.1454, 290.3952, 298.8429, 383.1141, 394.5482, 415.7989, 474.0785, 522.2687, 555.9868, 571.7233, 617.1713, 646.9524, 712.1052, 757.1555, 839.7896, 862.2479, 874.9923, 927.4888, 948.9697, 951.0036, 964.3596, 969.371, 1008.6015, 1039.7932, 1044.8249, 1063.0541, 1107.298, 1127.9082, 1155.2848, 1180.83, 1196.411, 1225.1961, 1234.4729, 1256.5558, 1278.3917, 1284.0116, 1311.6421, 1338.709, 1346.252, 1360.011, 1434.1602, 1439.0059, 1455.3892, 1490.6434, 1512.7327, 1517.3906, 1521.4376, 1525.9011, 1531.1185, 1540.3454, 1546.1395, 1554.7932, 1841.6486, 3045.7824, 3050.0779, 3053.1525, 3064.5046, 3070.2651, 3073.4956, 3094.2865, 3097.3753, 3101.0081, 3107.7236, 3108.5122, 3115.0888, 3117.7676, 3123.2296, 3127.9553, 3141.7127])
intens = np.array([3.609400e+00, 6.870000e-02, 1.425000e-01, 1.908000e-01, 2.848000e-01, 9.040000e-01, 7.114000e-01, 3.850000e-01, 1.899100e+00, 7.697000e-01, 1.484000e-01, 1.223400e+00, 5.366000e-01, 4.554700e+00, 2.007100e+00, 8.798000e-01, 9.361000e-01, 1.767700e+00, 4.380000e-01, 6.543100e+00, 4.705000e-01, 1.423900e+00, 5.475000e-01, 1.230200e+00, 3.059800e+00, 4.872000e-01, 1.293400e+00, 2.782900e+00, 5.430000e-02, 1.592800e+00, 2.582030e+01, 2.047560e+01, 1.544500e+00, 4.941600e+00, 1.135200e+00, 6.229000e-01, 3.967100e+00, 1.082100e+00, 5.126800e+00, 3.136400e+00, 3.190000e-02, 3.438700e+00, 6.669500e+00, 2.266600e+00, 1.033200e+00, 4.739000e+00, 4.292300e+00, 4.469500e+00, 6.858500e+00, 8.952200e+00, 2.593600e+00, 6.386200e+00, 4.342300e+00, 2.799900e+00, 1.920900e+00, 3.788000e-01, 4.900100e+00, 4.086800e+00, 2.093403e+02, 1.231370e+01, 1.935290e+01, 3.692450e+01, 2.791320e+01, 1.315910e+01, 2.868290e+01, 2.371370e+01, 1.425640e+01, 4.406400e+00, 7.293400e+00, 5.097790e+01, 4.594300e+01, 3.229710e+01, 1.685690e+01, 2.933100e+01, 2.938250e+01])

# 一次性计算所有75个峰，结果形状为(4001, 75)
all_gaussians = gaussian(intens, mu)

# 若需要叠加所有峰得到总光谱，直接按列求和
total_spectrum = all_gaussians.sum(axis=1)

修改后，x的形状变为(4001,1)，numpy会自动将mu和intens广播为(1,75)的形状，完美匹配运算需求，且计算速度远快于循环。

方案2：循环逐个独立计算 #

如果你希望严格按照"独立计算每个峰值"的逻辑来实现，可以通过循环逐个取出单个的强度和峰值位置进行计算：

import numpy as np

def gaussian(intens, mu):
    x = np.arange(4001)
    sig = 50
    return intens * np.exp(-np.power(x - mu, 2.) / (2 * np.power(sig, 2.)))

mu = np.array([106.2516, 169.2317, 179.4433, 210.1843, 225.1875, 237.6963, 261.1454, 290.3952, 298.8429, 383.1141, 394.5482, 415.7989, 474.0785, 522.2687, 555.9868, 571.7233, 617.1713, 646.9524, 712.1052, 757.1555, 839.7896, 862.2479, 874.9923, 927.4888, 948.9697, 951.0036, 964.3596, 969.371, 1008.6015, 1039.7932, 1044.8249, 1063.0541, 1107.298, 1127.9082, 1155.2848, 1180.83, 1196.411, 1225.1961, 1234.4729, 1256.5558, 1278.3917, 1284.0116, 1311.6421, 1338.709, 1346.252, 1360.011, 1434.1602, 1439.0059, 1455.3892, 1490.6434, 1512.7327, 1517.3906, 1521.4376, 1525.9011, 1531.1185, 1540.3454, 1546.1395, 1554.7932, 1841.6486, 3045.7824, 3050.0779, 3053.1525, 3064.5046, 3070.2651, 3073.4956, 3094.2865, 3097.3753, 3101.0081, 3107.7236, 3108.5122, 3115.0888, 3117.7676, 3123.2296, 3127.9553, 3141.7127])
intens = np.array([3.609400e+00, 6.870000e-02, 1.425000e-01, 1.908000e-01, 2.848000e-01, 9.040000e-01, 7.114000e-01, 3.850000e-01, 1.899100e+00, 7.697000e-01, 1.484000e-01, 1.223400e+00, 5.366000e-01, 4.554700e+00, 2.007100e+00, 8.798000e-01, 9.361000e-01, 1.767700e+00, 4.380000e-01, 6.543100e+00, 4.705000e-01, 1.423900e+00, 5.475000e-01, 1.230200e+00, 3.059800e+00, 4.872000e-01, 1.293400e+00, 2.782900e+00, 5.430000e-02, 1.592800e+00, 2.582030e+01, 2.047560e+01, 1.544500e+00, 4.941600e+00, 1.135200e+00, 6.229000e-01, 3.967100e+00, 1.082100e+00, 5.126800e+00, 3.136400e+00, 3.190000e-02, 3.438700e+00, 6.669500e+00, 2.266600e+00, 1.033200e+00, 4.739000e+00, 4.292300e+00, 4.469500e+00, 6.858500e+00, 8.952200e+00, 2.593600e+00, 6.386200e+00, 4.342300e+00, 2.799900e+00, 1.920900e+00, 3.788000e-01, 4.900100e+00, 4.086800e+00, 2.093403e+02, 1.231370e+01, 1.935290e+01, 3.692450e+01, 2.791320e+01, 1.315910e+01, 2.868290e+01, 2.371370e+01, 1.425640e+01, 4.406400e+00, 7.293