Matplotlib Colourmap из прозрачного - PullRequest
Новая
       87

Matplotlib Colourmap из прозрачного

1 голос
/ 18 июня 2020

Я новичок в Python и действительно изо всех сил пытаюсь получить разумную цветовую карту для своих данных.

Я рисую массивы 29x29 numpy, где большинство ячеек 0, но в среднем около 10-15 ячеек имеют ненулевые значения, которые могут варьироваться от 10 до нескольких тысяч.

В C ++ ROOT вы автоматически получаете красивый график с белым фоном и красивой цветовой полосой радуги, которую вы можете см. ниже.

Однако в matplotlib, следуя совету здесь:

python цветовая шкала тепловой карты matplotlib from transparent С кодом: 

from matplotlib.colors import LinearSegmentedColormap
%matplotlib inline 
#Lets visualise some events

# plot states
# plot states

# get colormap
ncolors = 256
color_array = plt.get_cmap('gist_rainbow')(range(ncolors))

# change alpha values
color_array[:,-1] = np.linspace(1.0,0.0,ncolors)

# create a colormap object
map_object = LinearSegmentedColormap.from_list(name='rainbow_alpha',colors=color_array)

# register this new colormap with matplotlib
plt.register_cmap(cmap=map_object)

# set colourbar map
cmap_args=dict(cmap='jet')

fig, axarr = plt.subplots(nrows=1, ncols=3)

axarr[0].imshow(events[0],**cmap_args)
axarr[0].set_title('Event0',fontsize=16)
axarr[0].tick_params(labelsize=16)

axarr[1].imshow(events[1],**cmap_args)
axarr[1].set_title('Event1',fontsize=16)
axarr[1].tick_params(labelsize=16)

axarr[2].imshow(events[2],**cmap_args)
axarr[2].set_title('Event2',fontsize=16)
axarr[2].tick_params(labelsize=16)

fig.subplots_adjust(right=2.0)
plt.show()

Я получаю изображения, подобные приведенному ниже, которые невозможно прочесть.

Пожалуйста, объясните, как получить белый фон и радужную цветовую полосу сбоку от графика?

Спасибо много!

text

text

1 Ответ

3 голосов
/ 18 июня 2020

Чтобы отображать все нулевые значения как белые, можно установить «под» цветом . Нижний цвет используется для значений, которые ниже самого низкого значения на шкале цветов. Если цветовая шкала должна начинаться с 1 с помощью vmin=1, все значения ниже 1 будут считаться «меньше». 

from matplotlib import pyplot as plt
import numpy as np
from matplotlib.ticker import MultipleLocator
#%matplotlib inline

# create a colormap object
cmap = plt.get_cmap('rainbow')
cmap.set_under('white')

# set colourbar map
cmap_args = dict(cmap=cmap, vmin=1, vmax=8000)

fig, axarr = plt.subplots(nrows=1, ncols=4, figsize=(12, 3), gridspec_kw={'width_ratios': [10, 10, 10, 1]})

events = np.random.randint(0, 9, size=(3, 10, 10)) * 1000 * np.random.randint(0, 2, size=(3, 10, 10))

for ax, event, title in zip(axarr[:3], events, ['Event 0', 'Event 1', 'Event 2']):
    img = ax.imshow(event, **cmap_args)
    ax.set_title(title, fontsize=16)
    ax.tick_params(labelsize=16)
    ax.xaxis.set_major_locator(MultipleLocator(1))
    ax.yaxis.set_major_locator(MultipleLocator(1))
fig.colorbar(img, cax=axarr[3])

# to make the colorbar exactly the same height as the image plots:
pos_ax2 = axarr[2].get_position()
pos_ax3 = axarr[3].get_position()
pos_ax3.y0 = pos_ax2.y0
pos_ax3.y1 = pos_ax2.y1
axarr[3].set_position(pos_ax3)

plt.show()

example plot

...