从散点图中的调色板中选择两种颜色

Question

我正在尝试使用只有两个值的 viridis 调色板在 python 中制作散点图。我真的很喜欢紫色，但黄色很明显。是否可以选择中间值（蓝色）和紫色？

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)

剧情：