Выбор двух цветов из палитры в точечной диаграмме

Question

Я пытаюсь построить точечный график в python, используя палитру viridis только с двумя значениями. Мне очень нравится фиолетовый, но желтый хорошо виден. Можно ли выбрать среднее значение (синий) и фиолетовый?

x_test = [1, 2, 3, 4]
y_test = [1, 2, 3, 4]
c_test = [0, 1, 0, 1]

plt.scatter(x = x_test, y = y_test, c = c_test,  alpha=1, cmap='viridis')

, и это дает два цвета - желтый и фиолетовый. Первый не виден.

Answer 1

Вы можете создать свою собственную цветовую карту только с 2 цветами:

Код:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors


x_test = [1, 2, 3, 4]
y_test = [1, 2, 3, 4]
c_test = [0, 1, 0, 1]

mycmap = colors.ListedColormap(['purple', 'blue'])

plt.scatter(x = x_test, y = y_test, c = c_test,  alpha=1, cmap=mycmap, s=100)

Сюжет: