머신러닝 파이썬

pandas value_counts()함수를 이용하여 bar차트 그리기

fast learner 2023. 5. 2. 09:24
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('../data_adp_book/student_data.csv')
print(df)
df.info()
plt.bar(df['grade'].value_counts().index, df['grade'].value_counts().values)
print(f"평균성적 {df['grade'].mean()}")
plt.show()

value_counts() 함수는 pandas의 Series 객체에서 사용할 수 있는 메서드로, 해당 Series에서 값들의 빈도수를 계산하여 반환합니다. 반환되는 값은 새로운 Series 객체이며, 이 객체의 인덱스는 고유한 값들이고, 해당 인덱스의 값은 각 값의 빈도수입니다.

plt.bar(df['grade'].value_counts().index, df['grade'].value_counts().values) 코드에서, df['grade'].value_counts()는 'grade' 열에 있는 값들의 빈도수를 계산합니다. 이 결과를 바탕으로 막대 그래프를 생성하기 위해 아래와 같이 사용되었습니다.

  1. df['grade'].value_counts().index: 이 부분은 'grade' 열의 고유한 값들을 나타냅니다. 이 값들은 막대 그래프의 x축에 위치합니다.
  2. df['grade'].value_counts().values: 이 부분은 각 고유한 값들의 빈도수를 나타냅니다. 이 값들은 막대 그래프의 y축에 위치하며, 해당 값들의 높이를 결정합니다.

따라서, plt.bar(df['grade'].value_counts().index, df['grade'].value_counts().values) 코드는 'grade' 열의 값들의 빈도수를 기반으로 한 막대 그래프를 생성하는 역할을 합니다.

728x90
반응형