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

Matplotlib에서 데이터 목록으로 히스토그램 만드는 방법

Matplotlib에서는 리스트 형태의 데이터를 활용해 간단하게 히스토그램을 그릴 수 있습니다. pyplot 모듈의 hist() 함수를 사용하면 되는데, 전체 과정은 다음 세 단계로 정리할 수 있습니다.

히스토그램 작성 단계

  • 데이터 준비: 히스토그램을 그릴 x 데이터 포인트를 리스트 형태로 생성합니다.
  • 히스토그램 그리기: plt.hist() 함수에 x 데이터를 전달해 히스토그램을 그립니다.
  • 그래프 표시: show() 메서드를 호출해 완성된 그래프를 화면에 출력합니다.

예제 코드

from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
x = [[300, 400, 500, 2000, 10], [300, 400, 500, 2000, 10], [300, 400, 500, 2000,
    10], [300, 400, 500, 2000, 10], [300, 400, 500, 2000, 10]]
plt.hist(x, 10)
plt.show()

코드 설명

rcParams 설정을 통해 그림 크기를 7.00 × 3.50 인치로 지정하고, figure.autolayout을 활성화해 레이블이나 제목이 잘리지 않도록 자동 조정했습니다.

여기서 사용된 x는 여러 개의 하위 리스트를 포함하는 중첩 리스트입니다. 이처럼 여러 데이터 시리즈를 함께 전달하면, Matplotlib는 각 시리즈별로 구분되는 막대를 한 화면에 겹쳐서 그려줍니다.

plt.hist(x, 10)에서 두 번째 인자인 10은 빈(bin)의 개수를 의미합니다. 즉, 전체 데이터 범위를 10개의 구간으로 나누어 각 구간에 속하는 값의 빈도를 계산하고 막대 높이로 표현합니다.

실행 결과

위 코드를 실행하면 각 데이터 시리즈가 10개의 구간으로 나뉘어 표시된 히스토그램이 출력됩니다.

Matplotlib에서 데이터 목록으로 히스토그램 만드는 방법