Python-Matplotlib：标准化y轴以显示标准差的倍数

Question

我想重新标准化我的y轴，以sigma（标准偏差）的倍数显示信号。例如，有人可以说在50Hz时有3 sigma信号，而在3Hz时有0.5 sigma信号。

我认为使用plt.yticks()是可行的方法：

import numpy as np
import matplotlib.pyplot as plt

X = range(0,50,2)
Y = range(0,50,2)

signal_sigma = np.std(Y)

plt.figure()
plt.plot(X, Y)
plt.yticks(np.arange(0, 25*signal_sigma, signal_sigma))
y_labels = [r"${} \sigma$".format(i) for i in range(0, 26)]
plt.ylabel(y_labels)
plt.show()

但这似乎还不太正确。我想念什么？

更新：

这就是我想做的：What does a 1-sigma, a 3-sigma or a 5-sigma detection mean?概率表正下方的位。

Answer 1

您要设置的yticklabels与使用plt.ylabel设置轴标签不同：

import numpy as np
import matplotlib.pyplot as plt

x = np.arange(1000)
y = 42 * np.random.randn(1000)

signal_sigma = y.std()

num_sigma = 3
sigma_values = np.arange(-num_sigma, num_sigma+1)
yticks = signal_sigma * sigma_values
yticklabels = ['$'+str(k)+'\sigma$' if k != 0 else '$\mu$' for k in sigma_values]

plt.figure()
plt.plot(x, y)
plt.yticks(yticks, yticklabels)
plt.ylabel('the axis label')