위와 같이 생긴 플롯을 컬러맵이라고 한다.

  • matplotlib.pyplot 모듈은 컬러맵을 간편하게 설정하기 위한 여러 함수를 제공한다.
  • plt.viridis(),plt.plasma(),plt.jet() 와 같은 함수를 이용해서 여러가지 색깔을 만들 수 있음.

컬러맵을 만들기 위해서는 아래의 함수를 사용해야 된다.

plt.subplot(nrows, ncols, index) - 그래프 내에 배열 형태로 나타내기 위해 사용

plt.scatter(x축값, y축값, c=기준축값) - x,y축을 넣어주고 어떤 축이 컬러 기준이 되는지 넣는다.
위 사진은 y축을 기준으로 했기 때문에 y축이 올라갈 수록 밝아진다.

plt.viridis() - 컬러맵 색상 지정(plt.plasma(),plt.jet() 얘네로 써서 스타일 바꿔도 됨.)

arr = np.random.standard_normal((8,100)) #난수 생성

plt.subplot(2,2,2)
plt.scatter(arr[0],arr[1],c=arr[1])
plt.viridis()

plt.title('viridis')

여기에 컬러바도 추가해주면 끝!

arr = np.random.standard_normal((8,100))

plt.subplot(2,2,2)
plt.scatter(arr[0],arr[1],c=arr[1])
plt.viridis()

plt.colorbar()
plt.title('viridis')

0개의 댓글