hungarian.py

import numpy as np
from scipy.optimize import linear_sum_assignment

def hungarian_method(cost_matrix):
    row_indices, col_indices = linear_sum_assignment(cost_matrix)
    return list(zip(row_indices, col_indices))

# Example usage:
cost_matrix = np.array([
    [3, 2, 7],
    [1, 4, 6],
    [5, 8, 9]
])

matching = hungarian_method(cost_matrix)
print(matching)