munkres_hungarian_algorithm_minimum_cost_matrix_assignment.py

1from munkres import Munkres, print_matrix
2
3matrix = [[5, 9, 1],
4          [10, 3, 2],
5          [8, 7, 4]]
6m = Munkres()
7indexes = m.compute(matrix)
8print_matrix(matrix, msg='Lowest cost through this matrix:')
9total = 0
10for row, column in indexes:
11    value = matrix[row][column]
12    total += value
13    print(f'({row}, {column}) -> {value}')
14print(f'total cost: {total}')