SolverStudio如何在for循环（PuLP）中引用2D列表中的1列

Question

我有2个数据集x1和x2。 我希望能够仅在“发件人”列中包含“奥克兰”的行中获得x1和x2的所有乘积的总和。 看这里

最终答案应该是（5 * 1）+（2 * 1）+（3 * 1）+（4 * 1）或14。下面给出了我为此编写的PuLP代码

# Import PuLP modeller functions
from pulp import *

varFinal = sum([x1[a] * x2[a] for a in Arcs if a == Nodes[0]])

print Nodes[0]
print Arcs[0]

Final = varFinal

打印到控制台的输出是

奥克兰市
（“奥克兰”，“阿尔巴尼”）

我意识到我的最终值为零，因为Arcs [some number]不等于Nodes [some number]。 无论如何，有没有更改代码，所以我的最终值为14？

任何帮助表示赞赏。

Answer 1

欢迎堆栈溢出！ 因为您只发布了部分代码，所以我不得不猜测您正在使用哪种数据类型。 从输出中，我猜您的Nodes是字符串，而您的Arcs是字符串的元组。

您的尝试非常接近，您希望from列中包含Auckland。 您可以像在数组中一样将其索引到元组中，因此您需要这样做： a[0] == Nodes[0] 。

以下是一个独立的示例，其中数据的第一位输出以下内容（请注意，我已更改为python 3.x打印语句（带括号））：

输出：

Auckland
('Auckland', 'Albany')
14

码：

# Import PuLP modeller functions
from pulp import *

# Data
Nodes = ['Auckland',
        'Wellington',
        'Hamilton',
        'Kansas City',
        'Christchuch',
        'Albany',
        'Whangarei',
        'Rotorua',
        'New Plymouth']

Arcs = [('Auckland','Albany'),
        ('Auckland','Hamilton'),
        ('Auckland','Kansas City'),
        ('Auckland','Christchuch'),
        ('Wellington','Hamilton'),
        ('Hamilton','Albany'),
        ('Kansas City','Whangarei'),
        ('Christchuch','Rotorua')]

x1_vals = [1, 2, 3, 4, 5, 9, 11, 13]
x2_vals = [5, 1, 1, 1, 1, 1, 1, 1]

x1 = dict((Arcs[i], x1_vals[i]) for i in range(len(Arcs)))
x2 = dict((Arcs[i], x2_vals[i]) for i in range(len(Arcs)))

varFinal = sum([x1[a] * x2[a] for a in Arcs if a[0] == Nodes[0]])

print(Nodes[0])
print(Arcs[0])
print(varFinal)

供您将来参考，如果您包含其他人可以尝试运行的代码（没有外部数据依赖项），那么人们很可能会给出答案，那样人们就可以尝试运行它，对其进行修复并重新发布。