不改变地块的y轴

Question

我所拥有的

我正在尝试使用python 3创建一个值的“热图”。

我困在y-滴答之间的间隔上了，有几个问题帮不了我.

到目前为止，我所掌握的代码如下：

data = np.load('path\to\data\datafile.npy')

# plot using imshow (and use interpolation)
plt.imshow(data,
           cmap          = 'RdBu_r', 
           aspect        = 'auto', 
           vmin          = -.4,
           vmax          =  .4,
           origin        = 'lower',
           extent        = [-1000, 4500, 2, 48], 
           interpolation = 'hanning')

# plt parameters
plt.xlim(-750, 4250)
plt.colorbar()
plt.vlines(0, 
           frequencies[0], 
           frequencies[-1], 
           colors = 'black',
           label = 'Stimulus onset',
           linewidth = 2.5)
plt.legend(loc=(1,1.04))
plt.title('Time-frequency using Morlet wavelets\nSubject: {}'.format(SUB_ID))
plt.ylabel('Frequency (in Hz)')
plt.xlabel('Time (in ms)')
plt.show()

这给了我以下的情节：

用错误的y轴标度绘图

我想要的

y轴上的标签是错误的.

具体来说，y轴应该有以下标签(以对数方式排列)：

In [15]:
# parameters
min_freq = 2;
max_freq = 48;
num_frex = 20;

# define frequencies of interest
frequencies = np.logspace(np.log10(min_freq), 
                          np.log10(max_freq), 
                          num_frex)
frequencies

Out[15]: 
array([ 2.        ,  2.36413729,  2.79457255,  3.30337659,  3.90481788,
        4.61576277,  5.45614844,  6.44954198,  7.62380134,  9.01185651,
       10.652633  , 12.59214343, 14.8847779 , 17.59482922, 20.7982959 ,
       24.58501342, 29.06117345, 34.35230187, 40.60677887, 48.        ])

生成的图形应该如下所示(当聚焦于y轴时)：

用正确的y标号间距绘制

所以：

• 情节不应改变。
• Y轴标签应从线性转换为日志。
• 标签之间的间距应与后一幅相同。

datafile.npy可以在这里找到

Answer 1

您只需要为情节提供一个自定义的ytick列表：

# parameters
min_freq = 2;
max_freq = 48;
num_frex = 20;

# define frequencies of interest
frequencies = np.logspace(np.log10(min_freq), 
                      np.log10(max_freq), 
                      num_frex)

plt.yticks(frequencies)

有关更多信息，请参见gen/matplotlib.pyplot.yticks.html#matplotlib.pyplot.yticks