Tạo một Legend colormap trong Matplotlib

Question

Tôi đang sử dụng imshow() trong matplotlib như vậy:

import numpy as np 
import matplotlib.pyplot as plt 
mat = '''SOME MATRIX''' 
plt.imshow(mat, origin="lower", cmap='gray', interpolation='nearest') 
plt.show() 

Làm thế nào để thêm một huyền thoại thể hiện các giá trị số cho các sắc thái khác nhau của màu xám. Đáng buồn thay, googling của tôi đã không phát hiện ra một câu trả lời :(

Cảm ơn bạn trước cho sự giúp đỡ

Vince

Answer 1

đơn giản, chỉ cần plt.colorbar():

import numpy as np 
import matplotlib.pyplot as plt 
mat = np.random.random((10,10)) 
plt.imshow(mat, origin="lower", cmap='gray', interpolation='nearest') 
plt.colorbar() 
plt.show() 

Answer 2

Như thường lệ, tôi hình dung nó ra ngay sau khi tôi hỏi nó;.). Đối với hậu thế, đây là lỗi của tôi tại đó:

m = np.zeros((1,20)) 
for i in range(20): 
    m[0,i] = (i*5)/100.0 
print m 
plt.imshow(m, cmap='gray', aspect=2) 
plt.yticks(np.arange(0)) 
plt.xticks(np.arange(0,25,5), [0,25,50,75,100]) 
plt.show() 

Tôi chắc chắn rằng có giải pháp thanh lịch hơn.

Vince

Answer 3

Có một BUILTIN colorbar() function trong pyplot.