이제 이 DataFrame에 있는 승차 위치를 점차트로 그리는 함수를 정의하자. 이때 뉴욕시의 핵심 랜드마크도 같이 그릴 수 있다. 구글로 검색해 보면 뉴욕시 주요 공항 두 개(JFK, 라과디아 공항)와 주요 지역의 좌표를 찾을 수 있다.
landmarks = {'JFK Airport': (-73.78, 40.643),
'Laguardia Airport': (-73.87, 40.77),
'Midtown': (-73.98, 40.76),
'Lower Manhattan': (-74.00, 40.72),
'Upper Manhattan': (-73.94, 40.82),
'Brooklyn': (-73.95, 40.66)}
그런 다음 맷플롯립을 사용해 승차 위치를 점차트로 그리는 함수를 정의한다.
import matplotlib.pyplot as plt def plot_lat_long(df, landmarks, points='Pickup'): plt.figure(figsize = (12,12)) # 차트 크기를 설정한다 if points == 'Pickup': plt.plot(list(df.pickup_longitude), list(df.pickup_latitude), '.', markersize=1) else: plt.plot(list(df.dropoff_longitude), list(df.dropoff_latitude), '.', markersize=1) for landmark in landmarks: plt.plot(landmarks[landmark][0], landmarks[landmark][1], '*', markersize=15, alpha=1, color='r') # 랜드마크를 지도 위에 표시한다 plt.annotate(landmark, (landmarks[landmark][0]+0.005, landmarks[landmark][1]+0.005), color='r', backgroundcolor='w') plt.title("{} Locations in NYC Illustrated".format(points)) plt.grid(None) plt.xlabel("Latitude") plt.ylabel("Longitude") plt.show()