我有使用 Google-OR Solver 在 Python 中解決分配問題的代碼。由于我是 Python 新手，因此在以通常的向量形式而不是復雜的結構域形式存儲變量時，我需要一些幫助。for i in range(0, assignment.NumNodes()):
print('Worker %d assigned to task %d. Cost = %d' % (i,assignment.RightMate(i), assignment.AssignmentCost(i)))我想要做什么我只想要'工人'和'任務'的兩個數組。對于工人很明顯，最后一個是 0，但我還需要任務索引的向量。我如何存儲這個'assignment.RightMate()'單個向量中的值并僅打印這些值
1 回答
達令說
TA貢獻1821條經驗 獲得超6個贊
使用列表推導獲取 assignment.RightMate() 的所有值
assignment_list = [assignment.RightMate(i) for i in range(assignment.NumNodes()]
對于工人，您只需：
worker_list = list(range(assignment.NumNodes()))
解釋
列表推導等效于以下 for 循環（這是執行此操作的另一種方法）：
assignment_list = [] for i in range(assignment.NumNodes()): assignment_list.append(assignment.RightMate(i))
注意 range() 默認從 0 開始，所以：
range(0, X) is equivalent to range(X)
添加回答
舉報
0/150
提交
取消