python - 每个点带有颜色编码值的散点图(如 pcolormesh)

Question

这个问题已经有答案了:

How to color scatter markers as a function of a third variable (4 个回答)

已关闭 5 年前。

我有由两个(非线性)键索引的数据。我正在尝试创建数据的散点，其中每个键位于一个轴上，并用一个代表数据值的颜色条。

颜色网格的问题是，我的数据看起来像，假设有 4 个点，(1,10,1st 值)，(1,20,2nd 值)，(2,13,3rd 值)和(2 ,14,4th 值)，所以我不能独立改变每个键。此外，没有数据的区域应保持白色。

我是 matplotlib 的新手，我还没有找到任何似乎能够做到这一点的函数。

我认为尝试通过修复几个值范围然后绘制散点图来手动创建色阶是可行的，但它看起来不优雅且不精确。

解决这个问题的最佳方法是什么，或者我错过的库中有相关的 matplotlib 函数/选项吗？

Answer 1

为了简单起见，我假设您可以获得第一个键的列表或数组，并将其命名为 x，第二个键并将其命名为 y，以及值并将其命名为值。

假设这一点，您可以结合seaborn和matplotlib来实现您正在寻找的东西:

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

x, y, value = np.random.rand(3, 50)
color_map = sns.dark_palette((0.5, 0.75, 0.9), as_cmap=True)

plot_handle, axis = plt.subplots()
scatter_points = axis.scatter(x, y, c=value, cmap=color_map)
plot_handle.colorbar(scatter_points)

# possibly need plt.show() if using interactively
# or else use matplotlib save figure options.

这会产生以下示例图:

您可以从 sns.palettes 中查看各种颜色映射选项(允许 as_cmap 的选项是最简单的)，包括一些不需要配置任何颜色范围，例如 sns.cubehelix_palette。