如何根据决策变量(一个用于行，一个用于列)从矩阵( Python中的列表列表)中挑选元素| OR-Tools，Python

from ortools.linear_solver import pywraplp

solver = pywraplp.Solver.CreateSolver('SCIP')

# 假设矩阵为matrix，行数为m，列数为n
x = []
for i in range(m):
    row = []
    for j in range(n):
        # 创建二进制变量，表示是否选择该元素
        var = solver.BoolVar('x[%i,%i]' % (i, j))
        row.append(var)
    x.append(row)

# 定义行约束条件，每行只能选择一个元素
for i in range(m):
    solver.Add(sum(x[i]) == 1)

# 定义列约束条件，每列只能选择一个元素
for j in range(n):
    solver.Add(sum(x[i][j] for i in range(m)) == 1)

# 假设目标函数为最大化选择的元素之和
objective = solver.Objective()
for i in range(m):
    for j in range(n):
        objective.SetCoefficient(x[i][j], matrix[i][j])
objective.SetMaximization()

status = solver.Solve()

if status == pywraplp.Solver.OPTIMAL:
    # 输出选择的元素
    for i in range(m):
        for j in range(n):
            if x[i][j].solution_value() == 1:
                print('选择元素[%i,%i]' % (i, j))
else:
    print('无可行解')