Python matplotlib : legend (범례 표시, 범례 표시하기, 항목 표시하기)

 

 

 

matplotlib의 legend method를 이용하면 chart에 범례를 표시할 수 있습니다.

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend()

plt.show()

위 코드에서처럼 legend method를 적어준 것 만으로도 범례가 표시되죠.

 

범례의 위치는 자동으로 적당한 위치에 나타나도록 결정됩니다.

 

 

 

 

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend(loc=(0.5, 0.5))

plt.show()

legend method에 loc 옵션은 범례 박스의 위치를 조절할 수 있도록 해줍니다.

 

(0.5, 0.5)의 의미는 범례의 위치를 의미하며, 결과로 나오는 차트의 x축 50%지점 y축 50% 지점에 범례 박스를 나타내라는 의미입니다.

 

더 정확히 말하면 범례의 왼쪽 하단 꼭지점이 x축 50%지점 y축 50% 지점에 오도록 범례 박스를 나타내라는 의미입니다.

 

 

 

 

 

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend(loc='upper left')

plt.show()

단순 좌표 뿐 아니라 lower right, lower left, upper right, upper left 과 같은 대략적인 위치를 전달할 수도 있습니다.

 

 

 

아래 표는 loc 옵션에 범례의 위치를 설정할 때 간편하게 사용할 수 있는 범례 위치 코드입니다.

Location String이나 Location Code 둘 중 아무거나 사용해도 되며 각각의 의미는 Description을 보면 됩니다.

Location String	Location Code	Description
'best'	0	그래프에서 가장 적절한 위치에 범례 표시 (default)
'upper right'	1	우측 상단에 범례 표시
'upper left'	2	좌측 상단에 범례 표시
'lower left'	3	좌측 하단에 범례 표시
'lower right'	4	우측 하단에 범례 표시
'right'	5	우측에 범례 표시
'center left'	6	좌측 가운데에 범례 표시
'center right'	7	우측 가운데에 범례 표시
'lower center'	8	가운데 하단에 범례 표시
'upper center'	9	가운데 상단에 범례 표시
'center'	10	가운데에 범례 표시

출처 = https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.legend.html

 

 

 

 

 

 

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend(ncol=3)

plt.show()

ncol 옵션은 범례를 나타낼 때 몇개의 열(column)이 있도록 나타낼지를 정합니다.

기본은 1개의 열에 여러 행을 나타내서 범례들을 표시하지만

위 예시에서처럼 ncol 옵션을 3으로 설정하면 3개의 열(column)되도록 범례를 표시합니다.

 

 

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend(frameon=True)

plt.show()

frameon 옵션을 True로 설정하면 범례 박스의 테두리가 나타납니다.

 

 

 

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend(fontsize=5)

plt.show()

fontsize 옵션을 이용하면 범례에 나타나는 글자의 크기 조절도 가능합니다.

 

 

 

 

 

 

import matplotlib.pyplot as plt

labels = ['Cake', 'Chocolate', 'Candy', 'Macaroon', 'Waffle']
values = [20, 35, 10, 50, 20]

plt.pie(values, labels=labels, colors=['skyblue', 'pink', 'grey', 'lightgreen', 'yellow'])
plt.legend(shadow=True)

plt.show()

shadow 옵션을 True로 하면 범례박스에 그림자가 표시되도록 할 수 있습니다.

 

 

 

 

 

 

 

 

 

import matplotlib.pyplot as plt

list_x = [1, 2, 3, 4, 5]
list_y = [2, 3, 4, 5, 6]

plt.plot(list_x, list_y, label='First line')
plt.legend()

plt.show()

 

legend method를 이용한 범례 표시는 다른 그래프에도 가능합니다.

위 예시는 선 그래프에 범례 표시를 한 예시입니다.

 

단, line chart는 pie chart처럼 그래프를 그릴 때 범례를 기본적으로 지정하지 않기 때문에 line chart의 label을 명시해줘야 합니다.

plot method에서 그래프를 그릴 때 label 옵션을 이용해서 원하는 line chart의 범례 이름이 뭔지 정해줄 수 있습니다.

 

 

 

 

 

import matplotlib.pyplot as plt

list_x = [1, 2, 3, 4, 5]
list_y_1 = [2, 4, 6, 8, 10]
list_y_2 = [2, 3, 4, 5, 6]

plt.plot(list_x, list_y_1, label='First line')
plt.plot(list_x, list_y_2, label='Second line')
plt.legend()

plt.show()

여러 개의 line chart가 동시에 존재한다고해도 각각의 line chart plot에서 label을 지정해주면 범례 표시가 가능합니다.