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

Python Pandas로 모든 열을 한 번에 표시하는 그룹화된 가로 막대 차트 그리기

Pandas에서 모든 열을 한꺼번에 보여주는 그룹화된 가로 막대 차트(grouped horizontal bar chart)를 만들려면 barh() 메서드를 사용하면 됩니다. 핵심은 x축과 y축 값을 별도로 지정하지 않는 것입니다. 이렇게 하면 DataFrame의 모든 수치 열이 자동으로 그룹화되어 하나의 차트에 함께 플롯됩니다.

1. 필요한 라이브러리 가져오기

먼저 데이터 처리를 위한 pandas와 시각화를 위한 matplotlib 라이브러리를 임포트합니다.

import pandas as pd
import matplotlib.pyplot as plt

2. DataFrame 생성하기

예제로 사용할 세 개의 열(Car, Cubic_Capacity, Reg_Price)을 가진 DataFrame을 만듭니다.

dataFrame = pd.DataFrame({
   "Car": ['Bentley', 'Lexus', 'BMW', 'Mustang', 'Mercedes', 'Jaguar'],
   "Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],
   "Reg_Price": [7000, 1500, 5000, 8000, 9000, 6000],
})

3. 그룹화된 가로 막대 차트 플롯하기

plot.barh()를 호출할 때 x, y 값을 지정하지 않고 제목(title)과 색상(color)만 설정합니다. 이 경우 문자열로 구성된 Car 열은 자동으로 인덱스(y축 레이블)로 사용되며, 나머지 수치 열인 Cubic_Capacity와 Reg_Price가 그룹으로 묶여 가로 막대로 표시됩니다.

dataFrame.plot.barh(title='Car Specifications', color=("blue", "orange"))

전체 예제 코드

아래는 위 과정을 모두 합친 완전한 코드입니다.

import pandas as pd
import matplotlib.pyplot as plt

# DataFrame 생성
dataFrame = pd.DataFrame({
   "Car": ['Bentley', 'Lexus', 'BMW', 'Mustang', 'Mercedes', 'Jaguar'],
   "Cubic_Capacity": [2000, 1800, 1500, 2500, 2200, 3000],
   "Reg_Price": [7000, 1500, 5000, 8000, 9000, 6000],
})

# 모든 열을 포함한 그룹화된 가로 막대 차트 플롯
dataFrame.plot.barh(title='Car Specifications', color=("blue", "orange"))

# 차트 화면에 표시
plt.show()

실행 결과

위 코드를 실행하면 각 자동차 모델(Bentley, Lexus, BMW 등)별로 배기량(Cubic_Capacity)과 등록 가격(Reg_Price)이 파란색과 주황색 막대로 나란히 비교되는 그룹화된 가로 막대 차트가 출력됩니다. 이 방식은 여러 지표를 카테고리별로 직관적으로 비교하고 싶을 때 특히 유용합니다.