[TA-Lib] #4: TA-Lib를 활용한 이동평균 분석 - 시장추세 확인 및 골든 크로스와 데드 크로스 포착

from yahooquery import Ticker
import talib
import pandas as pd

# 삼성 주식 데이터 받아오기
google = Ticker("005930.KS")
price_data = google.history(period="2y")
close_prices = price_data['close'].values

# 단순 이동 평균 계산
sma = talib.SMA(close_prices, timeperiod=20)

# 지수 이동 평균 계산
ema = talib.EMA(close_prices, timeperiod=20)

# 가중 이동 평균 계산
wma = talib.WMA(close_prices, timeperiod=20)

# 결과 출력
result = pd.DataFrame({
    'Close Prices': close_prices,
    'SMA': sma,
    'EMA': ema,
    'WMA': wma
})
print(result.tail())

     Close Prices      SMA           EMA           WMA
487       68900.0  70450.0  69944.221083  69709.047619
488       68000.0  70255.0  69759.057171  69475.714286
489       67500.0  69960.0  69543.908869  69213.333333
490       67300.0  69660.0  69330.203262  68960.000000
491       67000.0  69410.0  69108.279142  68706.666667

last_5_sma = sma[-5:]

# 상승 추세 확인
if all(last_5_sma[i] < last_5_sma[i + 1] for i in range(4)):
    print("시장은 상승 추세이다.")
# 하락 추세 확인
elif all(last_5_sma[i] > last_5_sma[i + 1] for i in range(4)):
    print("시장은 하락 추세이다.")
else:
    print("시장은 안정적이다.")

sma5 = talib.SMA(close_prices, timeperiod=5)
sma20 = talib.SMA(close_prices, timeperiod=20)
sma60 = talib.SMA(close_prices, timeperiod=60)

# 가정: 마지막 5일 동안의 단기, 중기, 장기 이동 평균 값
short_ma_last_5 = sma5[-5:]
mid_ma_last_5 = sma20[-5:]
long_ma_last_5 = sma60[-5:]

# 각 이동 평균이 연속적으로 상승하는지 확인
is_short_ma_rising = all(x < y for x, y in zip(short_ma_last_5, short_ma_last_5[1:]))
is_mid_ma_rising = all(x < y for x, y in zip(mid_ma_last_5, mid_ma_last_5[1:]))
is_long_ma_rising = all(x < y for x, y in zip(long_ma_last_5, long_ma_last_5[1:]))

# 추세 분석
if is_short_ma_rising and is_mid_ma_rising and is_long_ma_rising:
    print("강력한 상승 추세")
elif is_short_ma_rising:
    print("단기 상승 추세")
elif is_long_ma_rising:
    print("장기 상승 추세")
else:
    print("추세가 불분명함")

short_sma = talib.SMA(close_prices, timeperiod=20)
long_sma = talib.SMA(close_prices, timeperiod=60)

# 골든 크로스 찾기
for i in range(1, len(close_prices)):
    if short_sma[i] > long_sma[i] and short_sma[i-1] <= long_sma[i-1]:
        print(f"골든 크로스 발견: {i}일")

# 데드 크로스 찾기
for i in range(1, len(close_prices)):
    if short_sma[i] < long_sma[i] and short_sma[i-1] >= long_sma[i-1]:
        print(f"데드 크로스 발견: {i}일")