[암호화폐 자동매매] SMA (단순 이동 평균) 직접 구현 샘플

pandas를 이용하는 것으로 간주합니다. 아래는 SMA(단순 이동 평균)을 구하는 파이썬 코드입니다. 이미 구현되어 있는 라이브러리도 많지만 pandas를 활용한 방법은 공유합니다.

def sma(data, lookback, result_col_name):
    if 'close' not in data.columns:
        raise ValueError("The input DataFrame must contain a 'close' column.")
    if lookback <= 0:
        raise ValueError("The lookback period must be a positive integer.")
    
    data.loc[:, result_col_name] = data['close'].rolling(window=lookback).mean()
    return data

사용은 아래와 같이 합니다. sample은 pandas 객체이고 10일 이동평균값을 구하여 sma10 칼럼에 저장하도록 하는 코드입니다.

sample = sma(sample, 10, 'sma10')

pandas 객체에 SMA10 열이 생성되고 값이 저장됩니다.

끝.