[이동평균] #권트

이동평균.py

# 파라미터
# Data: 데이터(np.array), lookback: 이동 평균 기간(int), what: 대상 데이터 위치(int), where: 저장할 위치(int)
# ma: moving average
def ma(Data, lookback, what, where):
	for i in range(len(Data)):
		try:
			Data[i, where] = (Data[i - lookback + 1:i + 1, what].mean())
		except IndexError:
			pass
	return Data

# 사용 예시
# ma(대상 데이터, 이동평균 기간, 소스 칼럼, 타겟 칼럼)
# ma(ma_data, 100, CLOSE, MEAN)