ValueError：使用序列设置数组元素。 我可以在不改变一切的情况下解决它吗？

Question

我正在尝试在 python 中实现格子玻尔兹曼方法，为此我必须以这种方式定义这些变量：

ca = np.array([[0,0],[1,0],[0,1],[-1,0],[0,-1],[1,1],[-1,1],[-1,-1],[1,-1]]) # vitesses discrètes
w= np.array([4 / 9, 1 / 9, 1 / 9, 1 / 9, 1 / 9, 1 / 36, 1 / 36, 1 / 36, 1 / 36])
rho = np.ones((nx, ny))
ux = np.zeros((nx, ny))
uy = np.zeros((nx, ny))
radius = 0.5 * (1 - (R / R0) ** 2)
ux[:, :,0] = U0 - Y * (epsilon * U0 / R0) * np.exp(radius)
uy[:, :,0] = X * (epsilon * U0 / R0) * np.exp(radius)
geq = np.zeros((nx, ny, 9))

其中 nx=ny=100。 当我尝试以这种方式为每个索引从 0 填充到 8 时：

    geq[:, :, 0] = rho[:,:] * w[k] * (1 + (ca[0, 0] * ux[:, :] + ca[0, 1] * uy[:, :]) / c0 ** 2 + 0.5 * (
            ux[:, :] + uy[:, :]) ** 2 / c0 ** 4 - 0.5 * (ux[:,:]**2+uy[:,:]**2) / c0 ** 2)

我收到以下错误：

geq[:, :, k] = rho[:,:] * w[0] * (1 + (ca[k, 0] * ux[:, :] + ca[k, 1] * uy[:, :]) / c0 ** 2 +
ValueError: operands could not be broadcast together with shapes (110,100) (110,100,9) 

谁能告诉我如何解决上面损坏的代码中的问题，以便它停止抛出错误消息？ 我没有理解错误的含义。 我宁愿不修改变量的大小，只调整 for 中的行（如果可能的话）。