23 lines
706 B
Python
23 lines
706 B
Python
import pandas as pd
|
|
import matplotlib.pyplot as plt
|
|
|
|
class Util:
|
|
|
|
def analyze(self):
|
|
data = pd.read_csv("data_s.csv")
|
|
y_value = data.종가
|
|
|
|
max20 = y_value.rolling(window=20).mean()
|
|
stddev = y_value.rolling(window=20).std()
|
|
upper = max20 + (stddev * 2) # 상단 볼리저 밴드
|
|
lower = max20 - (stddev * 2) # 하단 볼리저 밴드
|
|
middle = (upper + lower) / 2 # 중심
|
|
|
|
x_value = [i for i in range(len(y_value))]
|
|
plt.plot(x_value, y_value.tolist(), x_value, upper.tolist(), x_value, lower.tolist(), x_value, middle.tolist())
|
|
plt.show()
|
|
return
|
|
|
|
if __name__ == "__main__":
|
|
util = Util()
|
|
util.analyze() |