1. 학습
    2. /
  2. 강의
    3. /
  3. Python에서 데이터 가져오기 입문
Connected

연습 문제

SQL 쿼리의 헬로 월드, 나만의 버전으로 커스터마이즈하기

첫 SQL 쿼리 실행을 축하해요! 이제 쿼리를 다음과 같이 커스터마이즈하는 방법을 알아보겠습니다:

  • 테이블에서 특정 컬럼 선택하기
  • 특정 개수의 행만 선택하기
  • 데이터베이스 테이블에서 컬럼 이름 가져오기

영상에서 Hugo가 아주 비슷한 방식으로 쿼리를 커스터마이즈했어요:

engine = create_engine('sqlite:///Northwind.sqlite')

with engine.connect() as con:
    rs = con.execute("SELECT OrderID, OrderDate, ShipName FROM Orders")
    df = pd.DataFrame(rs.fetchmany(size=5))
    df.columns = rs.keys()

다음과 같이 패키지는 이미 임포트되어 있습니다:

from sqlalchemy import create_engine
import pandas as pd

엔진도 이미 생성되어 있습니다:

engine = create_engine('sqlite:///Chinook.sqlite')

엔진 커넥션은 아래 구문으로 이미 열려 있습니다.

with engine.connect() as con:

완성해야 할 모든 코드는 이 컨텍스트 안에 작성하면 됩니다.

지침

100 XP
  • Employee 테이블에서 컬럼 LastName과 Title을 선택(SELECT) 하는 SQL 쿼리를 실행하세요. 결과는 변수 rs에 저장하세요.
  • rs에 fetchmany() 메서드를 적용해 레코드 3개를 가져오고, 이를 DataFrame df에 저장하세요.
  • rs 객체를 사용해 DataFrame의 컬럼 이름을 테이블의 컬럼 이름과 동일하게 설정하세요.