python - 如何在 MatPlotLib 中为散点图/绘图绘制平均线？

Question

我的数据如下：

x = [3,4,5,6,7,8,9,9]
y = [6,5,4,3,2,1,1,2]

我可以获得以下两个图表。

和

但是，我想要的是这个（沿途所有点的平均值）：

在matplotlib中可以吗？还是我必须手动更改列表并以某种方式创建：

x = [3,4,5,6,7,8,9]
y = [6,5,4,3,2,1,1.5]

相关代码

ax.plot(x, y, 'o-', label='curPerform')
x1,x2,y1,y2 = ax.axis()
x1 = min(x) - 1 
x2 = max(x) + 1
ax.axis((x1,x2,(y1-1),(y2+1)))

Answer 1

我认为这可以通过做最简单的方式来完成y_mean = [np.mean(y) for i in x]

示例：

import matplotlib.pyplot as plt
import random
import numpy as np


# Create some random data
x = np.arange(0,10,1)
y = np.zeros_like(x)    
y = [random.random()*5 for i in x]

# Calculate the simple average of the data
y_mean = [np.mean(y)]*len(x)

fig,ax = plt.subplots()

# Plot the data
data_line = ax.plot(x,y, label='Data', marker='o')

# Plot the average line
mean_line = ax.plot(x,y_mean, label='Mean', linestyle='--')

# Make a legend
legend = ax.legend(loc='upper right')

plt.show()

结果图：

Answer 2

是的，您必须自己进行计算。 plot绘制你给它的数据。如果您想绘制一些其他数据，您需要自己计算这些数据，然后再绘制它。

编辑：进行计算的快速方法：

>>> x, y = zip(*sorted((xVal, np.mean([yVal for a, yVal in zip(x, y) if xVal==a])) for xVal in set(x)))
>>> x
(3, 4, 5, 6, 7, 8, 9)
>>> y
(6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 1.5)