使用 Matplotlib 平滑折线图

Question

我有一个看起来像这样的 pandas 数据框：

import pandas as pd 
import matplotlib.pyplot

data = [{'A': 21, 'B': 23, 'C':19, 'D':26,'E':28,
        'F':26,'G':23,'H':22,'I':24,'J':21}] 

# Creates DataFrame. 
df = pd.DataFrame(data)

plt.figure(figsize=(12,8))

df.iloc[-1].plot(marker='o',markersize=5)

plt.show()

当我尝试在 Matplotlib 中使用 plot 时，我得到了一条非常锯齿状的线。

有没有办法让线条变得平滑，让它看起来更弯曲和流畅？

我曾尝试使用 scipy 的插值，但没有成功。

谢谢

Answer 1

这应该可以解决问题：

import pandas as pd 
import matplotlib.pyplot as plt
from scipy import make_interp_spline

data = [{'A': 21, 'B': 23, 'C':19, 'D':26,'E':28,
        'F':26,'G':23,'H':22,'I':24,'J':21}] 

# Creates DataFrame. 
df = pd.DataFrame(data)

y = np.array(df.iloc[-1].tolist())
x = np.arange(len(df.iloc[-1]))

xnew = np.linspace(x.min(), x.max(), 300) 
spl = make_interp_spline(x, y, k=3)
ysmooth= spl(xnew)

plt.plot(xnew, ysmooth)

Answer 2

这是一种选择（虽然不一定是理想的答案）：

您可以尝试对数据使用多项式近似值，但是您需要 x 轴和 y 轴的数值，我尝试了以下方法：

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

#note i've changed the A,B,C... to 1,2,3...
data = [{1: 21, 2: 23, 3:19, 4:26,5:28,
    6:26,7:23,8:22,9:24,10:21}] 

#Creates DataFrame. 
df = pd.DataFrame(data)

#define your lists
xlist = df.columns.tolist()
ylist = df.values.tolist()
ylist = ylist[0]

#plot data
plt.figure()
poly = np.polyfit(xlist,ylist,5)
poly_y = np.poly1d(poly)(xlist)
plt.plot(xlist,poly_y)
plt.plot(xlist,ylist)
plt.show()

另一种选择可能是样条插值，s 参数将允许您调整曲线的平滑度，您可以测试 s 的几个值：

from scipy.interpolate import splrep, splev

plt.figure()
bspl = splrep(xlist,ylist,s=25)
bspl_y = splev(xlist,bspl)
plt.plot(xlist,ylist)
plt.plot(xlist,bspl_y)
plt.show()