본문 바로가기
Data Science

[Python] MatPlotlib 기본 사용 방법

by 은구잇 2021. 11. 28.
728x90

1. MatPlotlib 라이브러리

matplotlib.pyplot 모듈은 그래프를 그릴 수 있는 파이썬 라이브러리

 

import matplotlib.pyplot as plt

plt.plot([1,2,3,4]) # 리스트의 값들이 y 값이라고 가정, [0,1,2,3] 을 자동으로 만들어냄
plt.show()  # 그래프를 화면에 나타냄

 

 

 

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(-15,10)
y1 = x **2
y2 = -x *20

plt.plot(x,y1,linestyle = "-.",marker = "*",color = "red",label = "y = x*x")
plt.plot(x,y2,linestyle = ":", marker = "o", color = "blue",label = "y = -x")

plt.xlabel("X")
plt.ylabel("Y")

# 그래프의 범례를 표시할 수 있음.
#plot 함수에 label 로 지정된 애들이 범례로 표시.
# legend(loc=(0.5,0.5)) , legend(loc = 'lower right')
plt.legend(
    loc = 'center',
    shadow = True,  # 그림자 효과
    borderpad = 1  # 범례의 크기
)
plt.show()

 

import matplotlib.pyplot as plt
import numpy as np

x = np.random.rand(10)
y = np.random.rand(10)

colors = np.random.randint(0,100,10)
sizes = np.pi*1000*np.random.rand(10)
plt.scatter(x,y,c=colors, s = sizes, alpha = 0.7)
plt.show()

 

import matplotlib.pyplot as plt
plt.plot([1,2,3,4],[2,3,4,5], linestyle = "-.",marker = "*",label = "PriceTEST")
plt.legend(
    borderpad = 1)
plt.show()

 

 

subplot을 사용하면 여러개의 그래프를 그릴 수 있음.

import matplotlib.pyplot as plt

import numpy as np

 

x = np.linspace(0,np*pi,*10, 500)

fig,axes = plt.subplot(2,2)

axes[0][0].plot(x,np.sin(x), linestyle = ":", marker = "*", color  = "green")

axes[0][1].plot(x,np.cos(x))

axes[1][0].plot(x,np.sin(x))

axes[1][1].plot(x,np.cos(x), linestyle = "--", marker  = "o", color = "black")

 

plt.savefig('save_png.png') # 그래프를 파일로 저장

plt.show()