파이썬(python) 시각화 Matplotlib 기초 (2)

위와 같이 pyplot모듈을 불러온 뒤 간단한 데이터를 입력하여 시각화 실습을 진행합니다.

누적 막대 그래프(Stacked bar chart)

누적 막대 그래프를 그리는 방법은 plt.bar를 입력하여 만든 막대 그래프에 plt.bar를 한 번 더 입력하여 기존 막대 그래프 위에 막대 그래프를 하나 더 쌓는 형식입니다. 이 때 두 번째 막대 그래프에 bottom매개변수를 사용해야 하며, 이 매개변수는 위쪽 막대 그래프가 시작하는 y축의 값을 설정합니다.

위 예시에는 bottom매개변수의 전달인자로 y가 사용되었으며, 출력된 그래프에서 위쪽 막대 그래프가 시작되는 값을 살펴보면 1,2,3,4로 y의 값인것을 확인할 수 있습니다.

누적 가로 막대 그래프(Stacked horizontal bar chart)

누적 가로 막대 그래프를 그리는 데에는 누적 막대 그래프에서 bottom매개변수가 사용된 것 처럼 left매개변수가 사용됩니다.

그룹화된 막대 그래프(Grouped bar chart)

하나의 눈금에 여러 막대그래프를 출력하고자 한다면 위와 같이 각 눈금에 위치할 막대별 x축의 위치를 조정하여 그릴 수 있습니다. 위 코드를 살펴보면, 각 눈금마다 왼쪽막대의 x축 위치(left_x_position)와 오른쪽 막대의 x축 위치(right_x_position)를 for문으로 계산한 뒤, plt.bar를 두 번 사용하여 막대 그래프를 그렸습니다.

plt.bar함수에 사용된 width 매개변수는 각 막대의 너비를 결정합니다.

그룹화된 가로 막대 그래프의 경우 그룹화된 막대 그래프와 동일하게 코딩하되 매개변수 width만 height으로 변경해주면 됩니다.

히트맵(Heatmap)

히트맵을 그리기 위한 데이터프레임을 위와 같이 코딩하고 pcolor함수로 히트맵을 그립니다. 또한 colorbar함수로 색상 정보를 함께 출력합니다 이 때, x축과 y축의 데이터가 무엇인지 표현하기 위하여 xlabel과 ylabel함수를 사용하여 x축과 y축의 이름을 표기해줍니다.

Leave a Comment