본문 바로가기

Python

정말 간단한 등고선 예제

import numpy as np
import matplotlib.pyplot as plt

lats = np.linspace(1, 5, 5) #1~5까지를 5로 나눔
lons = np.linspace(1, 4, 4) #1~4까지를 4로 나눔
lat, lon = np.meshgrid(lats, lons)

sss = np.array([[0, 0, 0, 0, 0],
              [2, 0, 0, 0, 0],
              [0, 0, 0, 0, 0],
              [0, 0, 0, 0, 0]])

plt.title("Contour plots")
plt.contourf(lat, lon, sss, alpha=.75, cmap='jet')
plt.contour(lat, lon, sss, colors='black')
plt.show()