Как построить график матрицы m x n в python



Вот такая ситуация:



У меня есть массив, представленный списком. Скажем:



a = [[1,0,1],[1,1,1]]


И я хотел бы построить его в некотором визуальном представлении (что-то вроде этого).



Пожалуйста, есть ли какой-то простой способ, как это сделать в python?



Правка:



Я следовал первому исходному коду в this и мой код выглядит так:



arr = zeros((len(matrix), len(matrix[0])))
for i in xrange(len(matrix)):
for j in xrange(len(matrix[i])):
arr[i][j] = matrix[i][j]
matshow(arr)
675   2  

2 ответов:

В то время как ответ @Zhenya работает, я бы использовал matshow, а не imshow

import pylab as plt
a = [[1,0,1],[1,1,1]]
plt.matshow(a)
plt.show()

Вам не нужно вызывать Аргумент interpolation, и оси по умолчанию являются целыми числами. Смотрите пример ниже:

Введите описание изображения здесь

Это работает для меня:

>>> a = [[1,0,1],[1,1,1]]
>>> import matplotlib.pyplot as plt
>>> plt.ion()
>>> fig, ax = plt.subplots()
>>> ax.imshow(a, interpolation='nearest')
<matplotlib.image.AxesImage object at 0x354ecd0>
>>> plt.draw()

Comments

    Ничего не найдено.