如何不直接替换代码变量值来获取函数对应指定坐标的输出值

方案1：从已生成的mod_G网格中直接索引取值 #

你的khmax和Ncmax都是固定步长生成的等距数组，只需要找到目标坐标在原数组中的索引，就能直接定位到mod_G的对应值：

# 目标坐标：(kh=0.1, Nc=0.1)
target_kh = 0.1
target_Nc = 0.1

# 查找对应索引，argmin适配目标值非精确匹配的场景
y_idx = np.argmin(np.abs(khmax - target_kh))
x_idx = np.argmin(np.abs(Ncmax - target_Nc))

# 直接获取对应值
target_mod_G = mod_G[y_idx, x_idx]
print(target_mod_G)

注意：因为你的meshgrid传参顺序是(Ncmax, khmax)，所以mod_G的行维度对应khmax的取值、列维度对应Ncmax的取值，索引顺序是[y_idx, x_idx]不要写反。

方案2：封装计算逻辑为独立函数 #

如果不需要生成整个网格、只需要查询单个点的取值，可以把计算逻辑封装成函数，直接传入目标坐标计算即可，不需要修改原数组的定义：

import numpy as np

def calc_mod_G(target_kh, target_Nc, N, matrix_C, j_node, x_j, hmax):
    sum_real = 0
    sum_imag = 0
    for l in range(0, N):
        delta_x = x_j(l+1) - x_j(j_node)
        sum_imag += matrix_C[j_node-1][l] * np.sin(target_kh * delta_x / hmax)
        sum_real += matrix_C[j_node-1][l] * np.cos(target_kh * delta_x / hmax)
    Aj_real = target_Nc * sum_real
    Aj_imag = target_Nc * sum_imag
    # 计算G的实部虚部
    G_imag = -Aj_imag + (2 * Aj_real * Aj_imag) / 2 - (3 * Aj_real**2 * Aj_imag - Aj_imag**3) + (4 * Aj_real * Aj_imag * (Aj_real**2 - Aj_imag**2))/24
    G_real = 1 - Aj_real + (Aj_real**2 - Aj_imag**2)/2 - (Aj_real**3 - 3*Aj_real*Aj_imag**2)/6 + ((Aj_real**2 - Aj_imag**2)**2 - 4*Aj_real**2*Aj_imag**2)/24
    mod_G = np.sqrt(G_real**2 + G_imag**2)
    return mod_G

# 调用示例
target_mod_G = calc_mod_G(target_kh=0.1, target_Nc=0.1, N=N, matrix_C=matrix_C, j_node=j_node, x_j=x_j, hmax=hmax)
print(target_mod_G)

这种方案的优势是不需要预生成整个500*500的网格，单个点计算的算力开销极小，也支持任意坐标点的查询，不需要受原数组步长的限制。

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