Смещение точек рассеяния для каждого массива для дельты х - PullRequest
0 голосов
/ 22 мая 2018

Я пытаюсь разобраться в сюжете, который на данный момент выглядит так: it's a scatterplot of a series of 17 different groups, each corresponding to a specific colour

Я пытаюсь выяснить, как сделать сдвиг для разных dx значений для каждого набора данных в таком режиме, что последний элемент серии остается в центре (в данном случае q ), тогда как первые 8 элементов (изот a до h) будет смещен влево, последние 8 вправо (от g до p), каждый для своего смещения.

У меня есть два массива (M и M1) формы (12,17) для каждого элемента в цикле , так что форма соответствует one цвету точек разброса.

    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')

1 Ответ

0 голосов
/ 22 мая 2018

Вы можете предварительно задать смещение и создать набор значений 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()

Результат: enter image description here

...