matplotlib: Lấy mảng bản đồ màu

Question

Tôi mới dùng matplotlib và bị kẹt trong các bản đồ màu.

Trong matplotlib làm cách nào để có được toàn bộ mảng màu RGB cho một bản đồ màu cụ thể, giả sử là "nóng". Ví dụ: nếu tôi ở MATLAB, tôi đã thực hiện điều này:

# in matlab 
c = hot(256); 
disp(c) 

Bất kỳ ý tưởng nào?

Answer 1

Bạn có thể tra cứu các giá trị bằng cách gọi colormap như một chức năng, và nó chấp nhận mảng NumPy để truy vấn nhiều giá trị cùng một lúc:

In [13]: cm.hot(arange(256)) 
Out[13]: 
array([[ 0.0416 , 0.  , 0.  , 1.  ], 
     [ 0.05189484, 0.  , 0.  , 1.  ], 
     [ 0.06218969, 0.  , 0.  , 1.  ], 
     ..., 
     [ 1.  , 1.  , 0.96911762, 1.  ], 
     [ 1.  , 1.  , 0.98455881, 1.  ], 
     [ 1.  , 1.  , 1.  , 1.  ]]) 

Answer 2

Got it! Vì vậy, bạn chỉ cần đi trong cửa sổ lệnh của Matlab của bạn và gõ

CMAP = colormap (nameOfTheColormapYouWant)

thể colormap trong Matlab là: parula, máy bay phản lực, hsv, nóng, mát mẻ, mùa xuân, mùa hè, mùa thu, mùa đông , màu xám, xương, đồng, màu hồng, dòng, colorcube, lăng kính, cờ.

Bạn nhận được ma trận trong đó mỗi hàng là mã màu được sử dụng cho bản đồ màu.