lapx_linear_assignment_with_jonker_volgenant_lapjv.py

1import numpy as np
2import lap
3
4# Create a random cost matrix
5cost_matrix = np.array([
6    [4, 1, 3],
7    [2, 0, 5],
8    [3, 2, 2]
9])
10
11# Solve the linear assignment problem
12# lap.lapjv returns (cost, row_to_col_assignments, col_to_row_assignments)
13cost, x, y = lap.lapjv(cost_matrix, extend_cost=True)
14
15print(f"Minimum cost: {cost}")
16print(f"Row assignments (which column for each row): {x}")
17print(f"Column assignments (which row for each column): {y}")