天天看點

MACD、K線、均線 實作

MACD、K線、均線 實作
import pandas as pd 
import matplotlib.pyplot as plt 
import pymysql
import sys

from mpl_finance import candlestick2_ochl
from matplotlib.ticker import MultipleLocator

# 計算ema
def calEMA(df,term):
    for i in range(len(df)):
        if i==0: #第一天
            df.loc[df.iloc[i].name,'EMA'] = df.loc[df.iloc[i].name,'close']
        if i>0:
            df.loc[df.iloc[i].name,'EMA'] = (term-1)/(term+1) * df.loc[df.iloc[i-1].name,'EMA'] + 2/(term+1) * df.loc[df.iloc[i].name,'close']
    EMAlist = list(df['EMA'])
    return EMAlist

# 計算macd
def  calMACD(df,shortTerm=12,longTerm=26,DIFTerm=9):
    shortEMA = calEMA(df,shortTerm)