92일차 코딩학원(pandas, csv 문제. 특정 열 값을 추출하고 그래프로 출력하기)

능력단위 평가

1. 판다스를 이용하여 외부 파일(sales_data.csv)을 읽어들이고, 해당 파일의 판매 데이터를 활용하여 시각화 결과가 나오게 구현하시오. 

 

문제 풀이

sales_data_.csv

0.00MB

import pandas as pd
import matplotlib.pyplot as plt

# CSV파일 불러오기
sales_data1 = pd.read_csv('./sales_data_.csv')

# 각 열 출력하기
sales_data2 = sales_data1[['month', 'tv', 'laptop', 'phone']]
print(sales_data2)

# 각 열 추출하기
a = sales_data2['month']
print(a)

b = sales_data2['tv']
print(b)

c = sales_data2['laptop']
print(c)

d = sales_data2['phone']
print(d)

# 특정 열 값만 추출하기
a_val = a.values
print(a_val)

b_val = b.values
print(b_val)

c_val = c.values
print(c_val)

d_val = d.values
print(d_val)

# 리스트로 변경하기
a_list = a_val.tolist()
print(a_list)

b_list = b_val.tolist()
print(b_list)

c_list = c_val.tolist()
print(c_list)

d_list = d_val.tolist()
print(d_list)

# 주어진 리스트들
a_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
b_list = [363, 549, 820, 865, 943, 618, 211, 373, 789, 82, 880, 102]
c_list = [1624, 2493, 2536, 1014, 1146, 1415, 1589, 737, 2789, 2180, 985, 1757]
d_list = [911, 1500, 1831, 1576, 1468, 835, 1117, 1998, 1441, 569, 1694, 599]

# 리스트들을 데이터프레임으로 변환
df = pd.DataFrame({'Month': a_list, 'TV Sales': b_list, 'Laptop Sales': c_list, 'Phone Sales': d_list})

# 그래프에 중첩해서 그리기. 마커 사용.
ax = df.plot(x='Month', y=['TV Sales', 'Laptop Sales', 'Phone Sales'], marker='o')

# 레이블 설정
ax.set_xlabel('Month')
ax.set_ylabel('Unit')

# 레전드 설정
ax.legend(['tv Sales', 'laptop Sales', 'phone Sales'], loc='upper left')

# 눈금 설정
ax.set_xticks(a_list)  # x축 눈금 설정
ax.set_yticks([1000, 2000])  # y축 눈금 설정

# 제목 설정
ax.set_title('name Sales Data')

plt.show()

 

 

결과