Gekko环境下获取GK.Operator值及R数组异常问题求助

问题1：Q数组元素无法获取正确计算值 #

• 原因：Q[idx]定义的是Gekko Operator（表达式），大写的.VALUE是变量的初始值，而非求解后的计算结果。Gekko中，求解后的结果需通过小写的.value属性获取；直接对Operator使用.VALUE会返回初始默认值0，而非表达式的实际计算结果。
• 解决：模型求解后，改用.value获取Q元素的计算值，确保表达式结果被正确读取。

问题2：R数组求解后部分元素变为数组 #

• 原因：使用numpy.full初始化R矩阵后，通过R[idx] = model.FV(name='R')赋值时，model.FV()生成单个FV对象，赋值给多个索引位置会触发numpy的广播机制，导致部分位置变成包含FV对象的数组，而非直接的FV对象。
• 解决：改用Gekko内置的model.Array创建R矩阵，或循环为每个三角位置单独创建FV对象，避免numpy的广播问题。

修改后的完整代码 #

from gekko import GEKKO
from numpy import triu_indices, diag_indices

# 创建Gekko模型
model = GEKKO()

n = 2
idx = triu_indices(n, 1)

# 创建1xn的CV数组
T = model.Array(model.CV, n)

# 用Gekko Array创建R矩阵，避免numpy广播问题
R = model.Array(model.FV, (n, n))
# 初始化对角线为固定值0
for i in range(n):
    R[i,i].VALUE = 0
    R[i,i].STATUS = False  # 固定值，不参与求解
# 填充上三角及对称下三角
if n > 1:
    for i,j in zip(idx[0], idx[1]):
        R[i,j].VALUE = 4
        R[i,j].STATUS = False  # 固定值，若需优化可设为True
        R[j,i] = R[i,j]

# 创建Q矩阵存储表达式
Q = model.Array(model.Const, (n, n))
for i in range(n):
    Q[i,i].VALUE = 0
if n > 1:
    for i,j in zip(idx[0], idx[1]):
        Q[i,j] = (T[j] - T[i]) / R[i,j]
        Q[j,i] = -Q[i,j]

# 给T赋值初始值
T[1].VALUE = 25
T[0].VALUE = 12

print("求解前R矩阵：")
for row in R:
    print([x.VALUE if hasattr(x, 'VALUE') else x for x in row])

model.solve(disp=False)

print("\n求解后R矩阵：")
for row in R:
    print([x.value if hasattr(x, 'value') else x for x in row])

print("\nQ[0,1]计算结果：", Q[0,1].value)
print("T[1]值：", T[1].value, "T[0]值：", T[0].value, "R[0,1]值：", R[0,1].value)
print("R[0,1]类型：", type(R[0,1]), "Q[0,1]类型：", type(Q[0,1]))

关键改动说明 #

1. R矩阵创建：改用model.Array(model.FV, (n,n))生成矩阵，每个元素都是独立的FV对象，避免numpy赋值时的广播问题，求解后元素类型保持一致。
2. 结果获取：Q元素的计算结果通过.value（小写）属性读取，得到正确的3.25（即(25-12)/4的结果）。
3. 固定值设置：对R的对角线元素设置STATUS=False，固定其值为0，无需参与求解过程。

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