print_matrix библиотеки munkres python генерирует исключение для матрицы, содержащей нули

Question

Lowest cost through this matrix:
Traceback (most recent call last):
    File "muncre.py", line 8, in <module>
        print_matrix(matrix, msg='Lowest cost through this matrix:')
    File "/usr/lib/python2.7/dist-packages/munkres.py", line 730, in    print_matrix
        width = max(width, int(math.log10(val)) + 1)
ValueError: math domain error

Когда матрица содержит ноль в любой из строк, выдается вышеупомянутая ошибка. Как я могу это исправить?

Это кусок кода в Python:

from munkres import Munkres, print_matrix
matrix = [[6, 9, 1],
          [10, 9, 2],
          [0,8,7]]
m = Munkres()
indexes = m.compute(matrix)
print_matrix(matrix, msg='Lowest cost through this matrix:')
total = 0
for row, column in indexes:
    value = matrix[row][column]
    total += value
    print '(%d, %d) -> %d' % (row, column, value)
print 'total cost: %d' % total

Я установил библиотеку munkres, используя следующую команду в Ubuntu:

sudo apt-get установить python-munkres

Answer 1

Это действительно похоже на ошибку в библиотеке munkres. Print_matrix - это просто "удобная" функция, и я бы предложил подать отчет об ошибке, а пока просто заменить его на что-то вроде следующего (это просто их код с исправлением, чтобы не пытаться применить 0 или отрицательные числа к логарифм). То, что мы пытались сделать, это сделать так, чтобы каждый столбец располагался как можно большей ширины для числа. Обратите внимание, что если вы передадите отрицательные числа, это может привести к отклонению на 1, но, с другой стороны, если у вас отрицательные расходы, у вас могут возникнуть более серьезные проблемы.

def print_matrix(matrix, msg=None):
    """
    Convenience function: Displays the contents of a matrix of integers.
    :Parameters:
        matrix : list of lists
            Matrix to print
        msg : str
            Optional message to print before displaying the matrix
    """
    import math

    if msg is not None:
        print(msg)

    # Calculate the appropriate format width.
    width = 1
    for row in matrix:
        for val in row:
            if abs(val) > 1:
               width = max(width, int(math.log10(abs(val))) + 1)

    # Make the format string
    format = '%%%dd' % width

    # Print the matrix
    for row in matrix:
        sep = '['
        for val in row:
            sys.stdout.write(sep + format % val)
            sep = ', '
        sys.stdout.write(']\n')