python - 如何更改 PSD 图中的 yticks？

Question

Download Zip with signal.csv

我可以创建一个 psd 图，如下所示:

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

df = pd.read_csv('signal.csv')/1000
dt = df.iloc[1,0] - df.iloc[0,0]
data=df['1']
print(f'time delta: {dt*1e12:.0f} ps')
print(f'time: {(len(data)*dt*1e6):.3f} \u03BCs')


resolution = 2000
plt.psd(data,
        resolution,
        1/0.1,
        lw=1,
        color='red');
plt.xlim(0,5)
plt.xlabel('Frequency / GHz')
plt.ylabel('Power Spectral Density / dB/Hz')
plt.show()

如何更改其 ytickrate？当我尝试添加时:

plt.yticks(np.arange(-100,0,10))

整个图表变形了:

如何更改 psd 图，以便在不更改图的情况下以 10 表示 y 轴？

Answer 1

您只需再次设置 y-lim:

plt.ylim((-72,-25))

您可以调整限制以匹配您所需的输出。

编辑如果您想使其自动，您可以使用axs.get_ylim():

fig, axs = plt.subplots(figsize=(10, 10),constrained_layout=True)
axs.psd(data,resolution,1/0.1,lw=1,color='red')

axs.set_xlim(0,5)
axs.set_xlabel('Frequency / GHz')
axs.set_ylabel('Power Spectral Density / dB/Hz')

ylim=axs.get_ylim()
axs.set_yticks(np.arange(-100,0,10))
axs.set_ylim(ylim)