본문 바로가기

인공지능/딥러닝

데이터 시각화 도구 - Matplotlib 과 plot 함수! (1)

반응형
  • 딥러닝 모델을 만들기 위해서는 내가 사용하는 데이터에 대한 특성을 잘 파악할 필요가 있습니다 이 때 내 데이터의 수치와 범주에 대해서 특성들을 분석 할 때 그래프를 그려보곤 하는데요. 이 때 필요한 도구가 Matplotlib 라이브러리입니다. 그럼 Matplotlib을 이용해서 그래프를 그리는 방법을 한 번 배워보도록 하겠습니다!

데이터 시각화의 경우 Matplotlib 라이브러리를 이용한다.

plot 그래프 그리기

  • 다음은 plot 그래프를 그린 예제이다!
import matplotlib.pyplot as plt

plt.plot([-1, 0, 1], [1, 0, 1], color='#ffafcb')
plt.show()

Untitled

그래프 색을 지정해주는 방법!

  • 색을 지정해 줄 때는 color 인자를 사용하면 됩니다! 코드를 보세요!

    • 색상을 지정해 줄 때 16진수를 사용해 입력하세요

      Untitled

```python
import matplotlib.pyplot as plt

plt.plot([-1, 0, 1], [1, 0, 1], color='#000000')
plt.show()
```

![Untitled](%E1%84%83%E1%85%A6%E1%84%8B%E1%85%B5%E1%84%90%E1%85%A5%20%E1%84%89%E1%85%B5%E1%84%80%E1%85%A1%E1%86%A8%E1%84%92%E1%85%AA%20%E1%84%8F%E1%85%A9%E1%84%83%E1%85%B3%20dc2eed827267494d8da7b11c31545ab6/Untitled%202.png)

![Untitled](%E1%84%83%E1%85%A6%E1%84%8B%E1%85%B5%E1%84%90%E1%85%A5%20%E1%84%89%E1%85%B5%E1%84%80%E1%85%A1%E1%86%A8%E1%84%92%E1%85%AA%20%E1%84%8F%E1%85%A9%E1%84%83%E1%85%B3%20dc2eed827267494d8da7b11c31545ab6/Untitled%203.png)

위 그림은 코드 결과 입니다! 이렇게 우리는 이제 그래프의 색상을 지정해 줄 수 있습니다!

그래프 x 절편과, y 절편의 범위를 변경해주고 싶을 때!

  • matplotlib에는 axis 함수가 존재합니다! 이 axis 함수를 이용해서 그래프의 x절편 범위와, y절편 범위를 변경해 줄 수 있습니다.
```python
import matplotlib.pyplot as plt

x = np.arange(0, 20)
y = x * 2

plt.plot(x, y, color = '#00ff00')
plt.axis([-10, 10, -10, 10]) # x절편 범위 , y절편 범위
plt.show()
```

![Untitled](%E1%84%83%E1%85%A6%E1%84%8B%E1%85%B5%E1%84%90%E1%85%A5%20%E1%84%89%E1%85%B5%E1%84%80%E1%85%A1%E1%86%A8%E1%84%92%E1%85%AA%20%E1%84%8F%E1%85%A9%E1%84%83%E1%85%B3%20dc2eed827267494d8da7b11c31545ab6/Untitled%204.png)

- 그래프를 보시면 절편의 범위가 -10~10으로 바뀐 것을 보실 수 있습니다!

그래프 제목을 달아주고 싶을 때!

  • matplotlib에는 그래프에 제목을 지어주기 위해 title 함수를 이용합니다! 코드를 보시죠!

      import matplotlib.pyplot as plt
    
      x = np.arange(0, 20)
      y = x * 2
    
      plt.plot(x, y, color = '#00ff00')
      plt.axis([-10, 10, -10, 10]) # x절편 범위 , y절편 범위
      plt.title('Matplotlib Title!')
      plt.show()

    Untitled

    • 그림을 보시면 그래프 위에 제목이 생긴 것을 보실 수 있습니다!

각 그래프에 이름을 달아주고 싶을 때!

  • matplotlib에는 label 인자와, legend 함수를 통해서 각 함수의 그래프에 이름을 달아줄 수 있어요! 코드를 보시죠!

      import matplotlib.pyplot as plt
    
      x = np.arange(0, 20)
      y = x ** 2
    
      plt.plot(x, y, label='my_graph')
      plt.legend()
      plt.show()

    Untitled

    • 이런 식으로 이름을 붙여 여러 개의 그래프를 식별 할 수 있습니다!

plot 그래프 모양을 바꾸어 줄 수 있다!

  • matplotlib에서 plot 함수를 사용할 때 linestyle인자를 통해서 그래프 스타일을 변경해 줄 수 있어요! 코드로 확인해 볼게요!

      import math
      import matplotlib.pyplot as plt
    
      x = []
      y = []
      y2 = []
      for angle in range(360):
          x.append(angle)
          y.append(math.sin(math.radians(angle)))
          y2.append(math.cos(math.radians(angle)))    
    
      x1 = np.linspace(-2*np.pi, 2*np.pi, 360)
      cos = np.cos(x1)
    
      plt.plot(x1, cos,linestyle='--')
      plt.title('SINE WAVE AND COS WAVE')
      plt.show()

    Untitled

    • 그림을 보시면 점선으로 그래프의 스타일을 변경했습니다!

그래프에 x label과 y label 붙이기!

  • matplotlib에는 그래프에 x 절편과 y 절편 이름을 붙여 줄 수 있습니다! 코드를 보시죠!

      import matplotlib.pyplot as plt
    
      plt.xlabel('This is Xlabel')
      plt.ylabel('This is Ylabel')
      plt.show()

    Untitled

    • 그림을 보시면 절편에 이름이 생긴 것을 보실 수 있죠!

여기까지 선형 함수를 그리는 방법에 대해 알아보았습니다! 그럼 다음에는 BAR 그래프를 그리는 방법을 알아볼게요!

반응형

'인공지능 > 딥러닝' 카테고리의 다른 글

핸즈온 머신러닝 1장 연습문제 풀이!  (1) 2024.01.07
신경망 공부 - 트랜스포머?  (2) 2023.12.02
데이터 처리 도구 - 넘파이!  (0) 2023.11.10
페이스북 라마2 모델 다운받기  (0) 2023.10.21
NLP 기초  (0) 2023.10.18