기본 콘텐츠로 건너뛰기

[matplotlib]quiver()함수

시,고,저,종,거래량의 산출

일일 주가자료의 일부 기간의 값들을 추출한 뒤 시가, 고가, 저가, 종가, 그리고 거래량을 다시 계산 하기 위해 여러 방법이 있을 수 있다.
일반적으로 어떤 기간의 자료를 정리하기 위해서는 다음과 같이 산정한다.
시가: 그 기간이 시작되는 시초값(시가중 최초값)
고가: 그 기간의 고가중 최고값
저가: 그 기간의 저가중 최저값
종가: 그 기간에서의 최종값(종가들 중 최종값)
거래량: 그 기간의 거래량의 합(그 기간의 거래량의 평균치를 사용할 수도 있지만 다음 함수에서 거래량의 합을 사용)

이러한 계산을 위해 다음과 같이 UDF를 작성하였다.

def ohlcInGroupS(dat):
    day=dat.index[0]
    op=dat.ix[0,0]
    hp=max(dat.ix[:,1])
    lp=min(dat.ix[:,2])
    cp=dat.ix[len(dat)-1, 3]
    vol=sum(dat.ix[:,4])
    re=pd.DataFrame([op, hp, lp, cp, vol], index=['op', 'hp','lp','cp','vol'], columns=[day]).T
    return(re)

우선 이 지정된 기간의 인덱스로 그 기간이 시작되는 날짜를 사용한다. 위 함수의 처음 코드는 그 인덱스를 추출하기 위해 작성하였다. 
google finance로 부터의 자료 호출_python 포스트에서 소개한 방법으로 일정기간의 kospi 일일자료를 호출하여 사용한다. 

In [1]: from googlefinance.client import get_price_data, get_prices_data, get_prices_time_data
In [2]: p={'q':'KOSPI', 'x':'KRX', 'i':60*60*24, 'p': '1Y'}
In [3]: kos=get_price_data(p)

이 자료 중 2007-10월 자료를 위의 적용하여 보자. 

In [4]: kos10=kos.ix[(kos.index>='2017-10-01') & (kos.index<='2017-10-31'),:]
In [5]: print(kos10)
Open High Low Close Volume
2017-10-10 15:30:00 2425.63 2443.20 2421.10 2433.81 221744000
...
2017-10-27 15:30:00 2485.44 2499.15 2479.67 2496.63 511225000
2017-10-30 15:30:00 2513.87 2513.87 2497.42 2501.93 342006000
위 추출된 자료의 처음 인덱스(날짜)를 생성되는 데이터의 인덱스로 하자. 
In [6]: day=kos.index[0]
In [7]: day
Out[7]: Timestamp('2016-11-01 15:30:00')

다음의 max(), min()은 최대값과 최소값을 계산하기 위한 파이썬의 내장함수이다. 이 함수는 전달하는 객체의 모든 값들을 대상으로 한다. 즉, 행과 열로 구성되어 있는 numpy, 또는 pandas 구조의 행 또는 열을 구분하지 않는다. 그러므로 이 함수를 사용하기 위해서는 한 행 또는 한 열을 구분하여 대상객체로 전달하여야 한다. 각 행 또는 열의 값들을 별도로 계산하기 위해서는 np.max(), np.min() 함수를 사용한다. sum() 함수 역시 같은 방식으로 계산한다. np.sum()을 고려할 수도 있다.
이 함수 내의 len()은 객체의 길이를 나타내는데 행렬 구조일 경우는 행의 길이를 나타낸다.

In [8]: op=kos10.ix[0,0] In [9]: hp=max(kos10.ix[:,1]) In [10]: lp=min(kos10.ix[:,2]) In [11]: cp=kos10.ix[len(kos10)-1, 3] In [12]: vol=sum(kos10.ix[:,4])
In [13]: pd.DataFrame([op,hp, lp, cp, vol], index=['op', 'hp','lp','cp','vol'], columns=[day]).T
Out[13]: 
ophplpcpvol
2016-10-10 15:30:002425.632513.872421.12501.934.737824e+09

위 UDF에 적용하여 보자.

In [14]: ohlcInGroupS(kos10)

ophplpcpvol
2017-10-10 15:30:002425.632513.872421.12501.934.737824e+09

댓글

이 블로그의 인기 게시물

[Linear Algebra] 유사변환(Similarity transformation)

유사변환(Similarity transformation) n×n 차원의 정방 행렬 A, B 그리고 가역 행렬 P 사이에 식 1의 관계가 성립하면 행렬 A와 B는 유사행렬(similarity matrix)이 되며 행렬 A를 가역행렬 P와 B로 분해하는 것을 유사 변환(similarity transformation) 이라고 합니다. $$\tag{1} A = PBP^{-1} \Leftrightarrow P^{-1}AP = B $$ 식 2는 식 1의 양변에 B의 고유값을 고려한 것입니다. \begin{align}\tag{식 2} B - \lambda I &= P^{-1}AP – \lambda P^{-1}P\\ &= P^{-1}(AP – \lambda P)\\ &= P^{-1}(A - \lambda I)P \end{align} 식 2의 행렬식은 식 3과 같이 정리됩니다. \begin{align} &\begin{aligned}\textsf{det}(B - \lambda I ) & = \textsf{det}(P^{-1}(AP – \lambda P))\\ &= \textsf{det}(P^{-1}) \textsf{det}((A – \lambda I)) \textsf{det}(P)\\ &= \textsf{det}(P^{-1}) \textsf{det}(P) \textsf{det}((A – \lambda I))\\ &= \textsf{det}(A – \lambda I)\end{aligned}\\ &\begin{aligned}\because \; \textsf{det}(P^{-1}) \textsf{det}(P) &= \textsf{det}(P^{-1}P)\\ &= \textsf{det}(I)\end{aligned}\end{align} 유사행렬의 특성 유사행렬인 두 정방행렬 A와 B는 'A ~ B' 와 같...

[sympy] Sympy객체의 표현을 위한 함수들

Sympy객체의 표현을 위한 함수들 General simplify(x): 식 x(sympy 객체)를 간단히 정리 합니다. import numpy as np from sympy import * x=symbols("x") a=sin(x)**2+cos(x)**2 a $\sin^{2}{\left(x \right)} + \cos^{2}{\left(x \right)}$ simplify(a) 1 simplify(b) $\frac{x^{3} + x^{2} - x - 1}{x^{2} + 2 x + 1}$ simplify(b) x - 1 c=gamma(x)/gamma(x-2) c $\frac{\Gamma\left(x\right)}{\Gamma\left(x - 2\right)}$ simplify(c) $\displaystyle \left(x - 2\right) \left(x - 1\right)$ 위의 예들 중 객체 c의 감마함수(gamma(x))는 확률분포 등 여러 부분에서 사용되는 표현식으로 다음과 같이 정의 됩니다. 감마함수는 음이 아닌 정수를 제외한 모든 수에서 정의됩니다. 식 1과 같이 자연수에서 감마함수는 factorial(!), 부동소수(양의 실수)인 경우 적분을 적용하여 계산합니다. $$\tag{식 1}\Gamma(n) =\begin{cases}(n-1)!& n:\text{자연수}\\\int^\infty_0x^{n-1}e^{-x}\,dx& n:\text{부동소수}\end{cases}$$ x=symbols('x') gamma(x).subs(x,4) $\displaystyle 6$ factorial 계산은 math.factorial() 함수를 사용할 수 있습니다. import math math.factorial(3) 6 a=gamma(x).subs(x,4.5) a.evalf(3) 11.6 simpilfy() 함수의 알고리즘은 식에서 공통사항을 찾아 정리하...

sympy.solvers로 방정식해 구하기

sympy.solvers로 방정식해 구하기 대수 방정식을 해를 계산하기 위해 다음 함수를 사용합니다. sympy.solvers.solve(f, *symbols, **flags) f=0, 즉 동차방정식에 대해 지정한 변수의 해를 계산 f : 식 또는 함수 symbols: 식의 해를 계산하기 위한 변수, 변수가 하나인 경우는 생략가능(자동으로 인식) flags: 계산 또는 결과의 방식을 지정하기 위한 인수들 dict=True: {x:3, y:1}같이 사전형식, 기본값 = False set=True :{(x,3),(y,1)}같이 집합형식, 기본값 = False ratioal=True : 실수를 유리수로 반환, 기본값 = False positive=True: 해들 중에 양수만을 반환, 기본값 = False 예 $x^2=1$의 해를 결정합니다. solve() 함수에 적용하기 위해서는 다음과 같이 식의 한쪽이 0이 되는 형태인 동차식으로 구성되어야 합니다. $$x^2-1=0$$ import numpy as np from sympy import * x = symbols('x') solve(x**2-1, x) [-1, 1] 위 식은 계산 과정은 다음과 같습니다. $$\begin{aligned}x^2-1=0 \rightarrow (x+1)(x-1)=0 \\ x=1 \; \text{or}\; -1\end{aligned}$$ 예 $x^4=1$의 해를 결정합니다. solve() 함수의 인수 set=True를 지정하였으므로 결과는 집합(set)형으로 반환됩니다. eq=x**4-1 solve(eq, set=True) ([x], {(-1,), (-I,), (1,), (I,)}) 위의 경우 I는 복소수입니다.즉 위 결과의 과정은 다음과 같습니다. $$x^4-1=(x^2+1)(x+1)(x-1)=0 \rightarrow x=\pm \sqrt{-1}, \; \pm 1=\pm i,\; \pm1$$ 실수...