티스토리 뷰

새롭게 옮긴 직장에서 간단한 일을 하나 맡게 되었는데 관련된 정보 하나 남겨보자 한다.

기존 코드가 postgreSQL DB에 접근해 데이터를 받아오는 것이 있었는데, 쿼리 결과(배열)에 index로 접근해 코드의 이해/유지보수가 쉽지 않았다. 이것을 hash(python 에서는 dictionary)와 비슷하게 칼럼명 데이터에 접근할 수 있게 변경하고 싶다. 

내 안에서 떠오른 방법은 두가지였다.

1. postgreSQL 패키지인 psycopg2.extra를 이용해 query 날리기.

2. pandas 를 이용해 query 날리기. pandas는 요즘 python으로 data 조작하는데 거의 필수적인 패키지가 아닐까 싶지만, 이 경우에는 query도 pandas로 날리도록 수정도 해야하고, 투머치인 것 같아 위의 1번 방법을 채용했다.


기존코드는 psycopg2 만을 import 하고 있었는데, 거기에 psycopg2.extra 도 같이 import 해주는 것과 간단한 코드 수정으로 완료.


기존처럼 index로도 접근할 수 있고, 칼럼명으로도 접근할 수 있다. 간단하다.

댓글
댓글쓰기 폼