[데이터분석] matplotlib : 막대그래프 간단하게 만들기

 

matplotlib 라이브러리를 이용하면 그래프를 쉽고 간단하게 만들 수 있는데요,

가장 많이 사용하는 막대그래프를 그리는 기본적인 방법에 대해서 설명하도록 하겠습니다.

 

 

1. 막대그래프(bar chart)

먼저, 막대그래프를 그리는 코드와 출력된 그래프를 먼저 보고 설명하도록 하겠습니다.

 

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(3)
item = ['coffee', 'tea', 'ade']
values = [2000, 4000, 5000]

plt.bar(x, values)
plt.xticks(x, item)
plt.grid(True)
plt.show()

 

 

import matplotlib.pyplot as plt

import numpy as np

먼저, 막대그래프를 그리기 위해서 matplitlib 라이브러리와, x축 값으로 입력할 array를 위해 numpy 라이브러리를 import 해줍니다. (matplotlib.pyplot 을 plt로 명명)

 

x = np.arange(3)

item = ['coffee', 'tea', 'ade']

values = [2000, 4000, 5000]

그리고나서, x축과 y축에 들어갈 데이터를 입력해야 하는데요, 각각 'item', 'values' 리스트에 입력했고, 두 리스트의 길이를 동일하게 해 주었습니다. 이때 데이터의 길이만큼 array도 만들어줘야 한답니다.('x=np.arange(3)')

 

plt.bar(x, values)

plt.xticks(x, item)

plt.grid(True)

plt.show()

데이터를 모두 입력했다면, 이제 그래프를 그려야 하는데요, 막대그래프를 그리기 위해 matplitlib에서 제공하는 bar() 함수를 사용하였고, 함수 안에 array 데이터와 y축 데이터를 각각 인자로 넣어주었습니다.

그리고나서 x축의 이름을 item 리스트에 있는 값으로 설정을 해주었습니다.

그다음, 그리드를 나타내게 하기 위해 .grid(True) 를 사용했습니다.

이제 plt.show() 를 하면 설정한 그래프가 출력됩니다.