Нежелательная цветовая палитра на участке - PullRequest
Новая
       55

Нежелательная цветовая палитра на участке

0 голосов
/ 27 ноября 2018

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

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

plt.close('all')
fig, axes = plt.subplots(nrows = 1, ncols = 1)
fig.set_facecolor('white')
axes.set_ylabel('$dz$ [$\AA$]')
axes.set_xlabel('Time [ns]')
axes.spines['right'].set_visible(False)
axes.spines['top'].set_visible(False)
axes.yaxis.set_ticks_position('left')
axes.xaxis.set_ticks_position('bottom')
axes.tick_params(direction='out')
#axes.set_title('N/A')
axes.set_ylim(-20,10)
axes.set_xlim(0, 90)
cmap = plt.get_cmap('plasma')
colors = [cmap(i) for i in np.linspace(0, 1, 9)]
# Make Color Bar ------------------------------------------------------
cax = 0
divider = make_axes_locatable(axes)
cax = divider.append_axes('right', size='5%', pad=0.1)
im = axes.imshow(np.linspace(1, 8.5, 100).reshape(10, 10), cmap='plasma')
fig.colorbar(im, cax=cax)
#----------------------------------------------------------------------
for i, dist in enumerate(dz):
    if i % 100 == 0:
        x = i / 1000 
        y = dist
        phval = final_pH_array[i]
        axes.plot(x, y, 'k.', markersize = 4 , color = colors[int(phval)], clip_on = False)
plt.savefig('plot.pdf')
plt.show()

Результаты выглядят следующим образом:

enter image description here

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

Любая помощь с этим была бы великолепна!

1 Ответ

0 голосов
/ 27 ноября 2018

Я думаю im.set_visible(False) должно достичь того, что вы хотите.

Но, может быть, вам стоит взглянуть на plt.scatter.scatter возвращает PathCollection, который вы можете передать функции colorbar.

...