Как построить график матрицы 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)
2 ответов:
В то время как ответ @Zhenya работает, я бы использовал
matshow, а неimshowimport 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