如何解决Python代码中的ValueError: setting an array element with a sequence

原始代码 #

import numpy as np

n = 101                # Number of grid points
P = np.zeros((n, n))
xmin, xmax = 0.0, 1.0     # limits in the x direction
ymin, ymax = -0.5, 0.5

x = np.linspace(xmin, xmax, nx)
y = np.linspace(ymin, ymax, ny)
X, Y = np.meshgrid(x, y, indexing='ij')
a = ((D/(dx*dx))-(X-Y*Y)/(2*dx))
b = (2+X-4*D/(dx*dx))
c = ((D/(dx*dx))-(Y+X*Y)/(2*dx))
d = ((D/(dx*dx))+(Y+X*Y)/(2*dx))
e = ((D/(dx*dx))+(X-Y*Y)/(2*dx))

### boundary conditions
for i in range(len(P)):

     P[i, 0] = 0               # Left boundary
     P[i, -1] = 0                     # Right boundary
     P[0, i] = 0         # Lower boudary
     P[-1, i] = 0                    
f = 0.
w = 1.2

resid_crit = 1.0e-6

# Arbitrary value at the begining, higher than resid_crit
resid_ave = 100.

count = 0

while resid_ave > resid_crit:

    # Set at 0 so we can sum it up later
    resid_ave = 0.


    # Set count for points in checker-boarding
    cnt_points = 0

    # Loop over internal points only
    for i in range(1, n - 1):
        for j in range(1, n - 1):

            # Checker-boarding
            if ((i + j) % 2) == count % 2:
                # Residual
                residual = a[i,j] * P[i - 1, j] + b[i,j] * P[i, j] + c[i,j] * P[i, j - 1] + d[i,j] * P[i, j + 1] + e * P[
                    i+1, j] - f

                # Update psi value

                P[i, j] += -w * residual / b[i,j]

                # Update resid_ave as a sum of residuals
                resid_ave += abs(residual)


                cnt_points += 1

    #  Get average residual
    resid_ave = resid_ave / cnt_points

    # Print every 1000-dth residual
    if count % 1000 == 0:
        print("Residual: %.7f" % resid_ave)
        #print("Residual: ", resid_ave)

    # Count iterations
    count = count + 1

错误输出 #

TypeError: only size-1 arrays can be converted to Python scalars

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "C:\Users\gitsa\PycharmProjects\sagarPy1\SKtest.py", line 89, in <module>
    P[i, j] += -w * residual / b[i,j]
ValueError: setting an array element with a sequence.

问题原因 #

1. 核心错误：计算residual时，e是二维数组，但代码直接用e * P[i+1, j]，导致residual变成数组而非单个标量。P[i,j]是单个元素，无法接受数组赋值，触发ValueError。
2. 缺失变量定义：nx、ny、D、dx均未定义，会导致运行时提前报错。
3. 边界注释混淆：原代码中P[0,i]和P[-1,i]的注释与实际索引对应关系不符，易造成理解混乱。

修正后的代码 #

import numpy as np

n = 101                # Number of grid points
P = np.zeros((n, n))
xmin, xmax = 0.0, 1.0     # limits in the x direction
ymin, ymax = -0.5, 0.5
D = 0.1  # 补充扩散系数定义，可按需调整
nx = n   # 网格点数与nx一致
ny = n   # 网格点数与ny一致
dx = (xmax - xmin) / (n - 1)  # 计算x方向网格步长

x = np.linspace(xmin, xmax, nx)
y = np.linspace(ymin, ymax, ny)
X, Y = np.meshgrid(x, y, indexing='ij')
a = ((D/(dx*dx))-(X-Y*Y)/(2*dx))
b = (2+X-4*D/(dx*dx))
c = ((D/(dx*dx))-(Y+X*Y)/(2*dx))
d = ((D/(dx*dx))+(Y+X*Y)/(2*dx))
e = ((D/(dx*dx))+(X-Y*Y)/(2*dx))

### boundary conditions
for i in range(len(P)):
    P[i, 0] = 0               # Left boundary
    P[i, -1] = 0              # Right boundary
    P[0, i] = 0               # Upper boundary
    P[-1, i] = 0              # Lower boundary

f = 0.
w = 1.2
resid_crit = 1.0e-6

resid_ave = 100.  # 初始残差，大于临界值
count = 0

while resid_ave > resid_crit:
    resid_ave = 0.
    cnt_points = 0

    # 遍历内部网格点
    for i in range(1, n - 1):
        for j in range(1, n - 1):
            if ((i + j) % 2) == count % 2:
                # 修正e的索引，取当前点对应的e值
                residual = a[i,j] * P[i - 1, j] + b[i,j] * P[i, j] + c[i,j] * P[i, j - 1] + d[i,j] * P[i, j + 1] + e[i,j] * P[i+1, j] - f

                # 更新P[i,j]的值
                P[i, j] += -w * residual / b[i,j]

                resid_ave += abs(residual)
                cnt_points += 1

    # 计算平均残差
    resid_ave = resid_ave / cnt_points

    # 每1000次迭代打印一次残差
    if count % 1000 == 0:
        print("Residual: %.7f" % resid_ave)

    count += 1

关键修改点 #

• 补充缺失的D、nx、ny、dx变量定义，确保代码可正常运行。
• 将residual计算中的e * P[i+1, j]改为e[i,j] * P[i+1, j]，使residual成为单个标量，匹配P[i,j]的赋值要求。
• 修正边界条件注释，使其与实际索引对应关系一致，提升代码可读性。

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