Matplotlib에 컬러 바 표시

이 자습서에서는 Python의 matplotlib 그림에 컬러 바를 표시하는 방법에 대해 설명합니다.

컬러 바를 생성하려면matplotlib.pyplot.colorbar()함수를 사용해야합니다.

다음 코드는 이에 대한 간단한 예를 보여줍니다.

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

s = plt.scatter(s_x, s_y, c=s_x, cmap="viridis")

c = plt.colorbar()

위의 예에서 우리는 플롯 외부에 간단한 컬러 바를 만들었습니다. cmap매개 변수를 사용하여 컬러 맵을 지정했습니다.

컬러 바를 표시 할 축을 지정할 수도 있습니다. 원하는 경우 플롯의 축 위에 추가 할 수 있습니다.

예를 들면

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

fig, ax = plt.subplots()
cax = fig.add_axes([0.27, 0.8, 0.5, 0.05])

im = ax.scatter(s_x, s_y, c=s_x, cmap="viridis")
fig.colorbar(im, cax=cax, orientation="horizontal")

위의 예에서subplots()함수를 사용하여 figure 및 axes 객체를 가져오고이를 사용하여 컬러 바의 축을 생성했습니다. colorbar()함수에서cax매개 변수를 사용하여이를 지정했습니다.

또한 최종 색상 막대의 방향을 변경 한orientation매개 변수를 사용합니다. 이 외에도 다양한 매개 변수를 사용하여 컬러 바의 모양과 크기를 제어 할 수 있습니다. 예를 들어,shrink는 바의 측면 비율 인aspect작은 여백만큼 컬러 바의 크기를 줄일 수 있으며 더 많이 있습니다.