4주차 TIL - 데이터 시각화와의 원만한 협상

진짜 너무 힘들다...

import pandas as pd

import numpy as np

import matplotlib.pyplot as plt

file_path = 'manufacturing_data_400.csv'

df = pd.read_csv(file_path)

# 1) 라인별 그룹화 및 총합 계산

grouped = df.groupby('Line').sum().reset_index()

# 데이터 타입과 길이 확인

print(grouped['Line'])

print(grouped['Production'].values)

print(grouped['Defects'].values)

x = np.arange(len(grouped['Line']))  # 라인 이름을 정수 인덱스로 변환

production = grouped['Production']

defects = grouped['Defects']

fig, ax = plt.subplots()

# 두 가지 막대(Production/Defects)를 나란히 배치

width = 0.35

ax.bar(x - width/2, production, width, label='Production', color='blue')

ax.bar(x + width/2, defects, width, label='Defects', color='orange')

# 라인 이름을 x축에 설정

ax.set_xticks(x)

ax.set_xticklabels(grouped['Line'])

# 3) 그래프 꾸미기

plt.title('Total Production and Defects by Line')

plt.xlabel('Line')

plt.ylabel('Count')

plt.xticks(rotation=0)

plt.legend(['Production', 'Defects'])

plt.grid(axis='y')

plt.show()