为增量x的每个阵列移位散射点 [英] Shifting scatter points for each array for delta x

问题描述

我正在尝试整理一个目前如下所示的情节：

我正在尝试解决如何为每个数据集的不同dx值进行移位，以使系列的最后一项保持在中心(在本例中为q)，而前8项(从a到h)将向左移动，最后8项向右(从g到p)移动，每一项都有不同的偏移量。

对于循环中的每个项目，我有两个形状为(12，17)的数组(M和M1)，因此该形状对应于散射点的一种颜色。

    import matplotlib.cm as cm
    import matplotlib
    import matplotlib.pyplot as plt
    import numpy as np

    fig, ax0 = plt.subplots (nrows=1, ncols=1)

    months_expanded = np.zeros((12,17))  
    months = np.arange(1, 13)
    M  = np.random.rand(12,17)
    M1 = np.random.rand(12,17)
    datalist = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q']

    for idx, sub in enumerate(datalist):
        for i in range(0,12):
            months_expanded[i] = np.repeat(months[i], 17)

        difference = abs(M - M1)

        ax0.scatter(months_expanded[:, idx], difference[:, idx], label = sub)                       

        colormap = plt.cm.gist_ncar                                                 
        colorst = [colormap(i) for i in    np.linspace(0,0.9,len(ax0.collections))]       
        for t,j1 in enumerate(ax0.collections):
            j1.set_color(colorst[t])

       ax0.legend(loc='center left', bbox_to_anchor=(1, 0.5))
       ax0.yaxis.grid(True)
       ax0.xaxis.grid(True)

       matplotlib.rcParams.update({'font.size': 30})
    fig.set_size_inches(20,20)
    fig.savefig(outfile, bbox_inches='tight')

推荐答案

您可以预定义偏移量并创建一组包括该偏移量的x值。请参阅下面说明这一点的最小示例：

import numpy as np
import matplotlib.pyplot as plt

N = 5 # This would be 17 in your case

# Create offsets
dx = np.linspace(-0.5, 0.5, N)
# Put last data set in centre
dx[N//2:] += (dx[1]-dx[0])
dx[-1] = 0.0
dx_expanded = np.repeat(dx, 12).reshape(N, 12)

# X-values to be shifted
months = np.arange(1, 13)
months_expanded = np.tile(months, N).reshape(N, 12)
months_shifted = months_expanded + dx_expanded

# Y-values
M  = np.random.rand(12,N)
M = np.linspace(1, 5, 12)
M = np.tile(M, N).reshape(N,12)

# Plot results
for idx in range(N):
  plt.scatter(months_shifted[idx], M[idx])
plt.show()

结果：

这篇关于为增量x的每个阵列移位散射点的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！