cmap中每个种群的不同颜色

Question

假设我有四个数组，还有一个x和y数组

arr1
arr2
arr3
arr4
xarr
yarr

这4个数组中的每个数组都与我的x和y数组具有相同的长度，并且每个数组都有一个值，表示该种群在x和y中的特定点处的“强度”。我的计划是将其绘制为散点图，对散点图中具有最高值的每个点进行着色(因此，如果arr1在该特定点最强，我会将其涂成红色，如果arr2是最强，我会将其涂成蓝色，如果arr3是最强，我会将其涂成绿色，如果arr4是最强，我会将其涂成黑色。)

我已经可以做这个图(附在下面)。不要担心这里的标签或子图，我只是附加了这个，这样你就可以有一些视觉效果。我的问题是，我也想考虑力量的价值。我不知道该如何快速解释，所以请耐心等待。在附图中，我的颜色栏下面只有4种纯色，每种纯色代表一个单独的群体。我仍然希望颜色条有四种不同的颜色，但我希望这四种颜色中的每一种都有渐变，这样我仍然可以感觉到人口较强和较弱的区域-即使它仍然是所有四个区域中最强的。Matplotlib已经有一个类似这样的色彩映射(tab20b或tab20c)，但我只是不确定如何在我做事情的方式中实现它。

​

​

​

​

基本上，我是如何在x和y上绘制这个图的，在这个循环中，我比较每个总体，然后将我馈送到散点图( color )的颜色数组分配给0、1、2或3，这取决于谁是最强的：

       for i in range(0, len(xarr)):
          for j in range(0, len(yarr)):
              if arr1[i,j] > max(arr2[i,j], arr3[i,j], arr4[i,j]):
                  color = np.append(color, 0)
              elif arr2[i,j] > max(arr1[i,j], arr3[i,j], arr4[i,j]):
                  color = np.append(color, 1)
              elif arr3[i,j] > max(arr1[i,j], arr2[i,j], arr4[i,j]):
                  color = np.append(color, 2)
              else:
                  color = np.append(color, 3)

我现在想要找到一种方法，以我的颜色存储arri，j的值，同时仍然能够让每个不同的群体以不同的颜色显示。你有什么建议吗？我从来没有创建过自己的颜色栏，但我知道这是你可以做的事情。

Answer 1

如果您像在arr = np.array([arr1, arr2, arr3, arr4])中那样将所有数组放在一起，则可以计算maxs = arr.max(axis=0)形式的最大值，并创建过滤器来绘制每个点的部分。然后，可以用不同的颜色映射表绘制每个数组。

每个颜色映射表可以在单独的颜色条中可视化，可以通过子图进行定位。

import matplotlib.pyplot as plt
import numpy as np

M, N = 50, 30
arr = [np.random.randn(M, N).cumsum(axis=1).cumsum(axis=0).ravel() for _ in range(4)]
arr = np.array([arr_i - arr_i.mean() for arr_i in arr])
maxs = arr.max(axis=0)
x = np.tile(np.arange(M), N)
y = np.repeat(np.arange(N), M)

cmaps = ['Reds', 'Greens', 'Blues', 'Greys']
fig, axes = plt.subplots(ncols=5, figsize=(12, 4), gridspec_kw={'width_ratios': [30, 1, 1, 1, 1], 'wspace': 0.4})
for arr_i, cmap, cbar_ax in zip(arr, cmaps, axes[1:]):
    filter = arr_i == maxs
    scat = axes[0].scatter(x[filter], y[filter], c=arr_i[filter], cmap=cmap)
    plt.colorbar(scat, cax=cbar_ax)
plt.show()

​

​