如何在 Pyplot 中获得平滑的平均曲线

Question

因此 Tensorboard 具有很好的平滑功能，您可以在其中以阴影颜色绘制原始曲线，并在顶部以纯色绘制平滑版本：

我怎样才能用 Pyplot 做到这一点？此功能对于发现平滑曲线的趋势以及阴影原始曲线的方差至关重要。不过 Tensorboard 非常有限，所以我想在 matplot lib 中重新创建外观。有任何想法吗？

请注意，我不想在数据点之间“插值”，因为方差太大。

Answer 1

您可以使用rollingpandas的-object。通过这个，您可以定义一个特定大小的窗口，该窗口滚动您的数据，然后您可以使用它来计算平均值、总和或其他内容。

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

t = np.linspace(0, 1.5*np.pi, 2500)
y = np.sin(t**2)+np.random.random(2500)*.6
df = pd.DataFrame(y)
plt.plot(df[0], 'lightblue', df[0].rolling(10).mean(), 'b')