파이썬으로 배우는 알고리즘 트레이딩 정리

파이선 기초

설치

윈도우 7 기반, 파이썬은 아나콘다 배포판으로 설치, 아나콘다는 반드시 32비트용으로 설치해야 함. 국내 증권사 api 가 32비트 프로그램이기 때문인 것으로 사료됨. 책은 아나콘다 파이썬 3.6 32비트 버전임. 아나콘다 설치시 관리자 권한으로 설치.

파이썬 인터프리터 바탕화면에 바로가기 만들기.

C:\Anaconda3\Lib\idlelib\idle.bat 를 바탕화면에 바로가기 만듬. windows10에서는 c:\ProgramData\Anaconda3\Lib\idlelib\idle.bat 로 되어 있음.

변수

변수는 데이터의 주소를 담고 있음. 실제 데이터를 객체라고 함. 변수에 담긴 실제 데이터의 주소를 알고 싶으면 id(변수명).

>>> x = 100
>>> x
100
>>> id(x)
1845813952
>>> y = x
>>> y
100
>>> id(x), id(y)
(1845813952, 1845813952)

문자열

파이썬은 문자와 문자열을 구분하지 않음. 작은따옴표, 큰따옴표 모두 사용가능. 문자열은 리스트임. 리스트 사용문법과 같음.
len(문자열) : 문자 개수
문자변수명[시작위치:끝위치] : 문자열 일부만 자르기, 시작위치, 끝위치 중 하나는 생략가능. 시작위치 생략하면 0으로 간주, 끝위치 생략하면 마지막 위치로 간주.위치를 음수로 지정. 음수로 지정되면 맨마지막위치부터 오는쪽으로 가면서 -1, -2 ...

문자열 나누기

>>> my_jusik = "naver daum"
>>> my_jusik.split(' ')
['naver', 'daum']
>>> my_jusik.split(' ')[0]
'naver'
>>> split_jusik = my_jusik.split(' ')
>>> split_jusik
['naver', 'daum']
>>> split_jusik[0]
'naver'

문자열 합치기

>>> daum = "Daum"
>>> kakao = "Kakao"
>>> daum + ' ' + kakao
'Daum Kakao'

데이터 타입 확인하기

>>> type(x)
<class 'int'>
>>> type(id(x))
<class 'int'>
>>> type(4000)
<class 'int'>
>>> type(3.14159)
<class 'float'>
>>> type('안녕하새요')
<class 'str'>

리스트

>>> myList = [23,345,'sdfasd','fghfh',444]
>>> myList2 = []
>>> myList[3]
'fghfh'
>>> myList[-3]
'sdfasd'
>>> myList[1:3]
[345, 'sdfasd']

리스트 생성할 때 사용한 각괄호 와 인덱스의 곽괄호는 다른 개념임.

리스트의 인덱스 사용문법

각괄호내부의 숫자로 사용. [0], [3], [-2], [1:6], [3:-1]
양수 인덱스: 0 첫원소의 앞, 1 첫원소 뒤 둘째원소 앞, ...
음수 인덱스: -1 마지막원소 앞 마지막에서 둘째원소 뒤, ...

리스트에 데이터 삽입

리스트변수명.append('삽입될 데이터') : 리스트 마지막에 삽입됨
리스트변수명.insert(인덱스, '삽입될 데이터') : 인덱스 위치에 삽입됨

리스트데이터 삭제

del 리스트변수명[인덱스]

이차원리스트의 인덱싱

리스트변수명[2][4] 처럼 사용.

튜플

리스트는 각괄호사용, 튜플은 둥글괄호 사용.
리스트는 원소를 변경할 수 있지만, 튜플은 변경할 수 없다.
튜플은 대신 속도가 빠르다.
튜플원소에 접근하는 방법은 [인덱스]를 사용한다. 이를통해 [시작위치:끝위치]를 사용해 튜플원소 일부를 가져올 수 도 있음.

>>> a = (1,2,3,'하하하','호호호',4,5)
>>> a[2:4]
(3, '하하하')

딕셔너리

{키:값, 키:값 ...}
딕셔너리는 인덱스를 지원하지 않음. 인덱스가 아니라 키를 사용해야 함.

>>> aa = {1:10, 2:20, '삼':30}
>>> aa[1]
10
>>> aa['삼']
30
>>> aa[1:2]
Traceback (most recent call last):
  File "<pyshell#51>", line 1, in <module>
    aa[1:2]
TypeError: unhashable type: 'slice'

키값으로 숫자, 문자, 모두 사용가능.

딕셔너리 데이터 추가

딕셔너리변수명[키] = 데이터

딕셔너리 데이터 삭제

del 딕셔너리변수명[키]

딕셔너리 키 찾기, 값 찾기

>>> aa = {1:10, 0.3:30}
>>> aa[0.3]
30
>>> aa.keys()
dict_keys([1, 0.3])
>>> type(aa.keys())
<class 'dict_keys'>
>>> list(aa.keys())
[1, 0.3]
>>> aaKeys = list(aa.keys())
>>> aaKeys
[1, 0.3]
>>> aaValus = list(aa.values())
>>> aaValus
[10, 30]
>>> 0.3 in aa.keys()
True

파이썬 비교연산자

==, !=, >, <, <=, >=

if 문 사용문법

   if 조건식 :
        실행문
    elif:
        실행문
    else:
        실행문

for문 사용문법

for 변수명 in 리스트 :
    실행문

리스트 원소가 하나씩 변수명에 바인딩 되어 실행문을 반복함.

리스트 대신 range()이용 해도 됨

for 변수명 in range(1,10)
    실행문

여기서 range는 1 <= range(1,10) < 10 로 해석. 그러므로 변수명에는 1 부터 9까지 차레대로 바인딩 됨.

리스트 대신 튜플을 사용해도 됨.
for 변수명 in 튜플

for문에 딕셔너리 사용문법

for 키변수명,값변수명 in 딕셔너러변수명.ltems():
또는
for 키변수명 in 딕셔너리변수명.keys():
또는
for 값변수명 in 딕셔너리변수명.values():

while 문 사용법

while 조건문 :
    실행문

break 문 과 continue 문

루프 내부에 사용. break는 루프를 빠져 나가고, continue문은 그 다음 실행문을 실행하지 않고 계속 루프를 실행.

함수만들기

def 함수명(파라미터들):
    함수내 명령문
    return 반환값

모듈

모듈은 함수들과 데이터들을 모듈명.py 라는 파일로 만들어 놓은것.
이를 파이썬에서 임포트 하여 재사용 할 수 있음.
import 모듈명

임포트 된 모듈의 함수를 사용하고 싶으면
모듈명.함수명()

임포트 된 모듈의 변수를 사용하고 싶으면
모듈명.변수명

모듈을 제작하는 과정에서 모듈을 테스트 하기 위해 모듈 마지막에 다음과 같은 코드를 삽입한다.

...
if __name__ == "__main__"
    print(내부함수명(필요한파라미터))
    print(__name__)

이렇게 함으로써 모듈을 직접 실행 할 때는 위의 if문이 동작하여 함수가 제대로 작동하는지 볼 수 있고, import 할 때는 위의 if 문이 동작되지 않는다. 그 이유는 name 이라는 파이썬 내부변수가 이 모듈을 직접 실행한 경우 "main"으로 바인딩 되고 import 되었을 때는 파일명으로 바인딩 되기 때문이다.

Time 모듈

>>> import time
>>> current_time = time.ctime()
>>> print (current_time)
Thu May 30 22:02:02 2019
>>> dir(time)
['_STRUCT_TM_ITEMS', '__doc__', '__loader__', '__name__', '__package__', '__spec__', 'altzone', 'asctime', 'clock', 'ctime', 'daylight', 'get_clock_info', 'gmtime', 'localtime', 'mktime', 'monotonic', 'monotonic_ns', 'perf_counter', 'perf_counter_ns', 'process_time', 'process_time_ns', 'sleep', 'strftime', 'strptime', 'struct_time', 'thread_time', 'thread_time_ns', 'time', 'time_ns', 'timezone', 'tzname']
>>> time
<module 'time' (built-in)>
>>> import random
>>> random
<module 'random' from 'C:\\ProgramData\\Anaconda3\\lib\\random.py'>

dir(모듈명) 은 모듈에 적재된 데이터 및 함수들을 확인 할 수 있다.

import 후 모듈명을 적으면 모듈의 위치를 확인할 수 있다.

os모듈

디랙토리 에 관한 함수들 있다.
현재 디랙토리 구하기 os.getcwd()
디랙토리내 파일명 구하기 os.listdir(패스)
특정디랙토리내 실행파일 찾기

for x in os.listdir('c:\Anaconda3')
    if x.endswith('exe')
        print(x)

임포트하는 방법

import 모듈명
모듈명.변수명
import 모듈명 as 별명
별명.변수명
from 모듈명 import 함수명

함수명을 바로 사용할 수 있다. 다만 기존의 동일한 이름의 함수명을 중복되어 사라짐.

form 모듈명 import *

모듈내 모든 함수 및 데이터를 바로 사용할 수 있음. 다만 기존의 동일한 이름의 것들은 모두 중복되어 사라짐.

클래스

self : 클래스가 인스탄스화 될 때 그 인스탄스명 으로 바인딩 된다.
__init__ :생성자(매쏘드)
__del__ : 소멸자

정의

class 클래스명(상속받을_부모_클래스명):
    클래스변수명 = 0
    def __init__(self, 입력받는인스탄스변수명, ...):
        self.입력받는인스탄스변수명 = 입력받는인스탄스변수명
        self.다른인스탄스변수명 = "호호데이터"
        함수변수명 = "임의데이터"

    def 매쏘드명(self):
        print (self.인스탄스변수명)

   def __del__(self):

상속받을_부모_클래스명 은 생략가능. 상속 받는다면 부모 클래스에 존재하는 메소드를 그대로 사용할 수 있다.
클래스변수명은 인스탄스에서 호출할 수 있으며 모든 인스탄스가 공유한다. C에서 정적 변수랑 비슷
인스탄스변수는 매소드 안에서만 정의될 수 있음. 보통 생성자 안에서 정의함. 그러면 다른 매소드에서도 사용 가능. (단, self. 없이 그냥 변수명으로 정의되면 그 함수에서만 사용되고 소멸되는 로컬 변수임)
인스탄스변수는 인스탄스명.인스탄스변수명 처럼 호출되며, 인스탄스 마다 다른 값을 가진다.
메소드가 self 인자를 가지고 있으면 인스탄스로만 호출이 가능하고,
메소드가 self 인자가 없으면 클래스로만 호출이 가능
클래스명은 대문자로 시작이 관례

인스탄스란 ff = 클래스명() 처럼 클래스가 호출될 때
ff 를 인스탄스라 함.
메소드가 클래스로 호출된다는 것은 클래스명.메소드() 형식으로 호출하는 것을 말함.
메소드가 인스탄스로 호출된다는 것은 인스탄스명.메소드() 형식으로 호출하는 것을 말함.

class 에서 변수 사용 정리

클래스 내부에서 정의
: 클래스변수가 되고 인스탄스 되지 않고 클래스로 직접 호출 가능할 뿐만 아니라 인스탄스에서도 호출가능. java에서 정적변수처럼 동작

매소드 내부에서 self.변수명 처럼 정의
: 인스탄스 변수가 되고 인스탄스의 다른 메소드에서 호출 가능.

매소드 내부에서 self. 없이 변수명으로 정의
: 해당 메소드 내부에서만 호출가능.

PANDAS 사용

pasdas는 파이썬의 리스트와 딕셔너리의 개념을 혼합 짬뽕한 것. 매우 유용함.
임포트할 때 Series 와 DataFrame을 로컬네임스페이스 내로 직접 임포트 함.

from pandas import Series, DataFrame

시리즈는 리스트로 부터 변환된다.

kakao = Series([92600, 92400, 92100, 94300, 92300])
print(kakao)

이때 자동적으로 인덱스는 0을 포함한 정수형으로 생김

리스트로 부터 시리즈를 만들 때, 인덱스를 직접 지정할 수 있다.

시리즈변수명 = Series(데이터리스트, index=인덱스리스트)

예)

kakao2 = Series([92600, 92400, 92100, 94300, 92300], 
    index=['2016-02-19', '2016-02-18', '2016-02-17', '2016-02-16','2016-02-15'])

kakao2['2016-02-18'] 처럼 데이터에 접근할 수 있음.
인덱스 전체를 리스트로 얻으려면 kakao2.index
데이터 전체를 리스트로 얻으려면 kakao2.values

시리즈 + 시리즈 하면 같은 키값의 데이터 끼리만 더해짐.

DataFrame은 2차원 데이터 타입임.
딕셔너리로 DataFrame을 정의할 수 있다.

딕셔너리변수명 =    {'컬럼명' : 컬럼데이터리스트,
                    '컬럼명' : 컬럼데이터리스트,
                    '컬럼명' : 컬럼데이터리스트 ...}
데이터프레임변수명 = DataFrame(딕셔너러변수명)

예)

from pandas import Series, DataFrame
daeshin = {'open':  [11650, 11100, 11200, 11100, 11000],
           'high':  [12100, 11800, 11200, 11100, 11150],
           'low' :  [11600, 11050, 10900, 10950, 10900],
           'close': [11900, 11600, 11000, 11100, 11050]}
date = ['16.02.29', '16.02.26', '16.02.25', '16.02.24', '16.02.23']
daeshin_day = DataFrame(daeshin, columns=['open', 'high', 'low', 'close'], index=date)

DataFrame을 생성할 때 columes= 옵션을 써서 컬럼의 순서를 정의 할 수 있다. index= 옵션을 사용해서 인덱스를 지정할 수 있다.
컬럼을 얻어 올 땐, daeshin_day['close']
로우를 얻어 올 땐, daeshin_day.ix['16.02.29']
컬럼명을 리스트로 얻어 올 땐, daeshin_day.columes
인덱스를 리스트로 얻어 올 땐, daeshin_day.index

pyqt5

이벤트처리

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *

class MyWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("PyStock")
        self.setGeometry(300, 300, 300, 400)

        btn1 = QPushButton("Click me", self)
        btn1.move(20, 20)
        btn1.clicked.connect(self.btn1_clicked) # clicked 라는 이벤트가 발생할 때 실행할 함수를 지정함.

    def btn1_clicked(self):
        QMessageBox.about(self, "message", "clicked")

if __name__ == "__main__":
    app = QApplication(sys.argv)
    myWindow = MyWindow()
    myWindow.show()
    app.exec_()

pyqtDesign 으로 만든 ui 사용법

import sys 
from PyQt5.QtWidgets import * 
from PyQt5 import uic 
form_class = uic.loadUiType("main_window.ui")[0] 
class MyWindow(QMainWindow, form_class):  
    def __init__(self): 
        super().__init__() 
        self.setupUi(self) 

if __name__ == "__main__": 
    app = QApplication(sys.argv) 
    myWindow = MyWindow() 
    myWindow.show() 
    app.exec_()

SQLite

  1. sqlite3는 파이썬 표준 라이브러리 임. 별도의 설치과정 없이 바로 사용할 수 있음.
  2. 임포트 하기 import sqlite3
  3. 데이터베이스 커넥트
    • con = sqlite3.connect("파일경로및파일명")
    • 해당경로에 파일이 존재하지 않으면 새로 생성한다.
  4. Cusor 생성 cusor = con.cursor()
  5. sql명령어 실행 cusor.execute("sql명령어")
  6. 작업내용 db에 적용 con.commit()
  7. db 닫기 con.close()
  8. select 문을 execute 한 뒤 데이터 가져오기
    • 로우단위 가져오기 cursor.fetchone() : 리턴값은 튜플
    • 모든 데이터 가져오기 cursor.fetchall(): 리턴값은 리스트
  9. 판다스의 데이터프레임 df와 sqlite db 상호간 데이터 변환
    • 판다스에서 데이터프레임을 디비로 저장하고 읽을 수 있는 메소드 제공.
    • 판다스에서 db로 저장할 때 df.to_sql('테이블명', con)
    • db를 판다스로 읽을 때
      import pandas as pd
      from pandas import Series, DataFrame
      import sqlite3
      con = sqlite3.connect("sdfsdfs.db")
      df = pd.read_sql("select * from kakao", con, index_col='index')
      #db에 인덱스 컬럼이 있는 경우 해당 컬럼명을 쓴다. 없는 경우 none

키움증권 OpenAPI 사용법

OpenApi 연결, 키움이벤트 처리

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QAxContainer import *

class MyWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("PyStock")
        self.setGeometry(300, 300, 300, 150)

        self.kiwoom = QAxWidget("KHOPENAPI.KHOpenAPICtrl.1")  # OpenAPI kiwoom 객체 만들기
        self.kiwoom.dynamicCall("CommConnect()")              # API제공 함수 CommConnect() 호출하기

        self.text_edit = QTextEdit(self)
        self.text_edit.setGeometry(10, 60, 280, 80)
        self.text_edit.setEnabled(False)

        self.kiwoom.OnEventConnect.connect(self.event_connect) #OnEventConnect 이벤트 와 슬롯함수를 연결

    def event_connect(self, err_code):                         #OnEventConnect 이벤트 발생실 실행할 슬롯함수 정의
        if err_code == 0:
            self.text_edit.append("로그인 성공")

if __name__ == "__main__":
    app = QApplication(sys.argv)
    myWindow = MyWindow()
    myWindow.show()
    app.exec_()
  • self.API객체 = QAxWidget("OpenAPI_ProgID") : OpenAPI객체 만들기
  • API객체.dynamicCall("API함수명()") : API 너머 서버에게 주어진 함수를 통해 정보를 요구 응답이 return 값으로 올 수 있음.
  • 응답이 리턴값으로 바로 올 수도 있지만, 이벤트로 실행된 함수 (슬롯함수, 콜백함수)를 통해서 받아야만 되는 경우도 있다.
  • 이벤트 이름은 API에서 제공. 위 예에서는 OnEventConnect
  • 이벤트 발생시 실행할 함수를 구현할 때는 반드시 API에서 미리 정해 논 프로토콜을 지켜야 함. (따라서 API 매뉴얼 참조)

계좌정보 가져오기

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import  *
from PyQt5.QAxContainer import *

class MyWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # Kiwoom Login
        self.kiwoom = QAxWidget("KHOPENAPI.KHOpenAPICtrl.1")
        self.kiwoom.dynamicCall("CommConnect()")

        # OpenAPI+ Event
        self.kiwoom.OnEventConnect.connect(self.event_connect)

        self.setWindowTitle("계좌 정보")
        self.setGeometry(300, 300, 300, 150)

        btn1 = QPushButton("계좌 얻기", self)
        btn1.move(190, 20)
        btn1.clicked.connect(self.btn1_clicked)

        self.text_edit = QTextEdit(self)
        self.text_edit.setGeometry(10, 60, 280, 80)

    def btn1_clicked(self):
        account_num = self.kiwoom.dynamicCall("GetLoginInfo(QString)", ["ACCNO"])
        self.text_edit.append("계좌번호: " + account_num.rstrip(';'))

    def event_connect(self, err_code):
        if err_code == 0:
            self.text_edit.append("로그인 성공")

if __name__ == "__main__":
    app = QApplication(sys.argv)
    myWindow = MyWindow()
    myWindow.show()
    app.exec_()

종목 기본정보 가져오기

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import  *
from PyQt5.QAxContainer import *

class MyWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # Kiwoom Login
        self.kiwoom = QAxWidget("KHOPENAPI.KHOpenAPICtrl.1")
        self.kiwoom.dynamicCall("CommConnect()")

        # OpenAPI+ Event
        self.kiwoom.OnEventConnect.connect(self.event_connect)
        self.kiwoom.OnReceiveTrData.connect(self.receive_trdata)

        self.setWindowTitle("PyStock")
        self.setGeometry(300, 300, 300, 150)

        label = QLabel('종목코드: ', self)
        label.move(20, 20)

        self.code_edit = QLineEdit(self)
        self.code_edit.move(80, 20)
        self.code_edit.setText("039490")

        btn1 = QPushButton("조회", self)
        btn1.move(190, 20)
        btn1.clicked.connect(self.btn1_clicked)

        self.text_edit = QTextEdit(self)
        self.text_edit.setGeometry(10, 60, 280, 80)
        self.text_edit.setEnabled(False)

    def event_connect(self, err_code):
        if err_code == 0:
            self.text_edit.append("로그인 성공")

    def btn1_clicked(self):
        code = self.code_edit.text()
        self.text_edit.append("종목코드: " + code)

        # SetInputValue
        self.kiwoom.dynamicCall("SetInputValue(QString, QString)", "종목코드", code)

        # CommRqData
        self.kiwoom.dynamicCall("CommRqData(QString, QString, int, QString)", "opt10001_req", "opt10001", 0, "0101")

    def receive_trdata(self, screen_no, rqname, trcode, recordname, prev_next, data_len, err_code, msg1, msg2):
        if rqname == "opt10001_req":
            name = self.kiwoom.dynamicCall("CommGetData(QString, QString, QString, int, QString)", trcode, "", rqname, 0, "종목명")
            volume = self.kiwoom.dynamicCall("CommGetData(QString, QString, QString, int, QString)", trcode, "", rqname, 0, "거래량")

            self.text_edit.append("종목명: " + name.strip())
            self.text_edit.append("거래량: " + volume.strip())

if __name__ == "__main__":
    app = QApplication(sys.argv)
    myWindow = MyWindow()
    myWindow.show()
    app.exec_()

종목코드 및 한글종목명

import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import  *
from PyQt5.QAxContainer import *

class MyWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.kiwoom = QAxWidget("KHOPENAPI.KHOpenAPICtrl.1")
        self.kiwoom.dynamicCall("CommConnect()")

        self.setWindowTitle("종목 코드")
        self.setGeometry(300, 300, 300, 150)

        btn1 = QPushButton("종목코드 얻기", self)
        btn1.move(190, 10)
        btn1.clicked.connect(self.btn1_clicked)

        self.listWidget = QListWidget(self)
        self.listWidget.setGeometry(10, 10, 170, 130)

    def btn1_clicked(self):
        ret = self.kiwoom.dynamicCall("GetCodeListByMarket(QString)", ["0"])
        kospi_code_list = ret.split(';')
        kospi_code_name_list = []

        for x in kospi_code_list:
            name = self.kiwoom.dynamicCall("GetMasterCodeName(QString)", [x])
            kospi_code_name_list.append(x + " : " + name)

        self.listWidget.addItems(kospi_code_name_list)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    myWindow = MyWindow()
    myWindow.show()
    sys.exit(app.exec_())

댓글 남기기

이메일은 공개되지 않습니다. 필수 입력창은 * 로 표시되어 있습니다