在不改变绘图的情况下重新调整 y 轴

Question

我有什么

我正在尝试使用 python 3 创建值的“热图”。
我被困在 y 刻度之间的间距上，几个 SO 问题无法帮助我......

我目前的代码如下：

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

这给了我以下情节：

Plot with wrong y-axis scaling

我想要什么

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 轴时）：

Plot with correct y-label spacing

所以：

• 情节不应改变
• y 轴标签应从线性转换为对数
• 标签之间的间距应与后面的图中相同

datafile.npy can be found here

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)

更多信息请参见https://matplotlib.org/3.1.0/api/_as_gen/matplotlib.pyplot.yticks.html#matplotlib.pyplot.yticks