Python转写MATLAB矩阵计算报错：无法广播输入数组形状

我有一项作业需要将MATLAB程序转写为Python程序，该程序用于求解物体的振幅。尝试计算数组y时，始终报错：ValueError: could not broadcast input array from shape (50,9) into shape (50,)，不确定是y的定义问题还是存在其他遗漏，报错出现在最后一行。

报错代码 #

x = np.array([[1,1,1],[2,2,2],[3,3,3]])
z = np.array([[1,2,3],[1,2,3],[1,2,3]])
o = np.array([[0,0,2],[1,0,0],[0,0,0]])
flattenmatrixO = np.array(o).flatten('F')
N = len(flattenmatrixO)

xt = np.array([1,2,3])
zt = 0
L = len(xt)
K = 50
fs, Dt, c = 10, 1/10, 15e-1
y = np.zeros([K,N])

for i in range(0,L):
    tof = ((z-zt) + np.sqrt((x-xt[i])**2 + (z-zt)**2))/c
    kd = np.round(tof/Dt)
    kd = kd.astype(int)
    kd = np.array(kd).flatten(order='F')
    
    Hl = np.zeros([K,N])
    for n in range(0,N):
        Hl[kd[n],n] = 1
    H = [H,Hl]
    O = np.array(o).flatten(order='F')
    y[:,i] = Hl * O

MATLAB原代码对应行 #

y(:,i) = Hl * o(:)

问题原因与解决方案 #

1. 核心错误：
   Python中*是逐元素乘法，而MATLAB中*是矩阵乘法。代码里Hl * O得到(50,9)形状的数组，但y[:,i]是(50,)的一维数组，形状不匹配导致广播错误。

2. 修正步骤：

   • 替换乘法方式：用NumPy的@运算符或np.dot()实现矩阵乘法，这和MATLAB的*行为一致。
   • 调整y的维度：原代码中y = np.zeros([K,N])，但循环仅赋值前3列（L=3），改成y = np.zeros([K,L])更贴合逻辑。
   • 移除未定义变量：H = [H,Hl]第一次循环会报错，若无需该变量可直接删除，若需要则先初始化H = []再追加元素。

修正后的代码 #

import numpy as np

x = np.array([[1,1,1],[2,2,2],[3,3,3]])
z = np.array([[1,2,3],[1,2,3],[1,2,3]])
o = np.array([[0,0,2],[1,0,0],[0,0,0]])
flattenmatrixO = np.array(o).flatten('F')
N = len(flattenmatrixO)

xt = np.array([1,2,3])
zt = 0
L = len(xt)
K = 50
fs, Dt, c = 10, 1/10, 15e-1
y = np.zeros([K,L])  # 调整为K行L列，匹配循环次数

for i in range(0,L):
    tof = ((z-zt) + np.sqrt((x-xt[i])**2 + (z-zt)**2))/c
    kd = np.round(tof/Dt)
    kd = kd.astype(int)
    kd = np.array(kd).flatten(order='F')
    
    Hl = np.zeros([K,N])
    for n in range(0,N):
        # 确保kd[n]在合法索引范围内，避免越界
        if 0 <= kd[n] < K:
            Hl[kd[n],n] = 1
    O = np.array(o).flatten(order='F')
    # 用矩阵乘法替换逐元素乘法
    y[:,i] = Hl @ O  # 等价于np.dot(Hl, O)

额外提示 #

• 需确保kd中的值在0到K-1之间，否则会触发数组索引越界错误，可添加判断语句规避。

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