Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Matplotlib로 이산 값 히스토그램 그리기

개요

Matplotlib에서 이산 값(discrete values)에 대한 히스토그램을 그리려면 hist() 메서드를 사용하면 됩니다. 이산 데이터는 개별적인 값들로 구성되어 있기 때문에, 각 값이 고유한 구간(bin)에 대응되도록 bins 옵션을 데이터의 길이로 설정하는 것이 핵심입니다.

구현 단계

  • Figure 크기를 설정하고 서브플롯 주변과 사이의 여백(padding)을 조정합니다.
  • 이산 값으로 이루어진 리스트를 준비합니다.
  • hist() 메서드를 사용하여 bins=데이터 길이, edgecolor='black' 옵션과 함께 데이터를 시각화합니다.
  • 그래프를 화면에 표시하려면 show() 메서드를 호출합니다.

예제 코드

from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

data = [1, 4, 2, 3, 5, 9, 6, 7]
plt.hist(data, bins=len(data), edgecolor='black')
plt.show()

코드 설명

rcParams["figure.figsize"]는 Figure의 가로·세로 크기를 인치 단위로 지정하고, rcParams["figure.autolayout"]을 True로 설정하면 레이블이나 제목이 잘리지 않도록 여백이 자동으로 조절됩니다.

히스토그램을 그릴 때 bins=len(data)로 설정하면 데이터의 개수만큼 구간이 생성되어, 각 이산 값이 별도의 막대로 표현됩니다. 또한 edgecolor='black'을 지정하면 막대 경계가 검은색으로 표시되어 인접한 막대들을 쉽게 구분할 수 있습니다.

실행 결과

위 코드를 실행하면 각 이산 값(1, 4, 2, 3, 5, 9, 6, 7)이 개별 막대로 나타난 히스토그램이 출력됩니다. 검은색 테두리 덕분에 각 값의 빈도를 한눈에 파악할 수 있습니다.