将平均移动平均曲线安装到曲线的问题

Question

我在Python中的移动平均功能有问题。我尝试使用几种方法，但它们没有起作用。我在光线曲线上使用良好工作之前使用的移动平均值，但分阶段发生问题： 具有拟合移动平均值的相图

我不知道为什么它不正常。这是我使用的代码：

import matplotlib.pyplot as plt
import numpy as np

data = np.genfromtxt('phase.txt', dtype='float,float', names=["Time","Flux"])
t = data["Time"]
flux = data["Flux"]

def movingaverage(interval, window_size):
   window = np.ones(int(window_size))/float(window_size)
   return np.convolve(interval, window,'same')

av = movingaverage(flux, 10)

plt.plot(t,flux,"C0.", ms=2)
plt.plot(t,av,"r")
plt.show

我希望这样的东西： 拟合的移动平均线

这是我的文件，其中有两个列中的值： bpky4/查看？usp =共享

Answer 1

我注意到您的时间值和相应的通量值未正确排序。时间值应该上升。

import matplotlib.pyplot as plt
import numpy as np

data = np.genfromtxt('phase.txt', dtype='float,float', names=["Time","Flux"])
t = data["Time"]
flux = data["Flux"]

# sort time values in ascending order
sort_idx = t.argsort()
t = t[sort_idx]

# sort flux values accordingly
flux = flux[sort_idx]


def movingaverage(interval, window_size):
   window = np.ones(int(window_size))/float(window_size)
   return np.convolve(interval, window,'same')

av = movingaverage(flux, 10)

plt.plot(t,flux,"C0.", ms=2)
plt.plot(t,av,"r")
plt.ylim([0.95, 1.05]) # zoom in a little
plt.show()

这会产生您期望的结果吗？

---

我曾尝试过同时对时间值进行排序以及分类时间和磁通值。

仅分类时间值：

排序时间值和磁通值：

您看起来更合理？

Answer 2

您的数据非常分散。考虑在较大的窗口上平均。

import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('phase.txt', sep=' ', names=["Time","Flux"]).sort_values(by='Time')
df['MA_10'] = df.Flux.rolling(window=10).mean()
df['MA_100'] = df.Flux.rolling(window=100).mean()

plt.style.use('ggplot')
plt.figure(figsize=(16,9))
plt.scatter(df.Time, df.Flux, c='blue')
plt.plot(df.Time, df.MA_10, c='red')
plt.plot(df.Time, df.MA_100, c='black')
plt.ylim([0.95, 1.05])
plt.show()