Python Matplotlib에서 여러 줄 그리기
이 튜토리얼에서는 Matplotlib를 사용하여 여러 라인을 플로팅하고 그림의 각 라인에 다른 색상을 설정하는 방법을 설명합니다.
Python Matplotlib에서 한 줄 그리기
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(8)
y = 4 * x - 10
plt.plot(x, y)
plt.title("Plot line in Matplotlib", fontsize=15)
plt.xlabel("X", fontsize=13)
plt.ylabel("Y", fontsize=13)
plt.show()
출력:
matplotlib.pyplot.plot()함수를 사용하여 선의 플롯을 만듭니다. 줄의X와Y 좌표를plot()함수에 인수로 전달합니다.
Python Matplotlib에서 여러 줄 그리기
Matplotlib에서 여러 줄을 플롯하려면 각 줄에 대해matplotlib.pyplot.plot()함수를 계속 호출하고 해당 줄의 좌표를 각plot()함수에 인수로 전달합니다.
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(8)
y1 = 4 * x + 5
y2 = 3 * x + 5
y3 = 2 * x + 5
y4 = x + 5
plt.plot(x, y1)
plt.plot(x, y2)
plt.plot(x, y3)
plt.plot(x, y4)
plt.title("Plot Multiple lines in Matplotlib", fontsize=15)
plt.xlabel("X", fontsize=13)
plt.ylabel("Y", fontsize=13)
plt.show()
출력:
각기 다른 색상을 가진 공통 축으로 4 개의 다른 선을 그립니다. matplotlib.pyplot.plot()함수를 4 번 호출하여 4 개의 다른 선을 플로팅합니다. 함수에 대한 인수로 다른 라인의 좌표를 전달할 때마다.
플롯을보다 설명 적으로 만들기 위해 플롯에 범례를 추가하여 어떤 선이 어떤 색으로 표시되는지 추론 할 수 있습니다.
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(8)
y1 = 4 * x + 5
y2 = 3 * x + 5
y3 = 2 * x + 5
y4 = x + 5
plt.plot(x, y1, label="4x+5")
plt.plot(x, y2, label="3x+5")
plt.plot(x, y3, label="2x+5")
plt.plot(x, y4, label="x+5")
plt.title("Plot Multiple lines in Matplotlib", fontsize=15)
plt.xlabel("X", fontsize=13)
plt.ylabel("Y", fontsize=13)
plt.legend()
plt.show()
출력:
범례와 함께 그림에 4 개의 선을 표시합니다. 그림에 범례를 추가하기 위해plot()함수에서label 매개 변수를 설정하여 선을 그리는 동안 각 선에 대한 레이블을 설정합니다. 마지막으로matplotlib.pyplot.legend()를 호출하여 Figure의 범례를 추가합니다.
기본적으로 Matplotlib는 자동으로 선에 색상을 할당합니다. 각 라인의 색상을 제어하려면matplotlib.axes.Axes.set_prop_cycle()메서드를 사용합니다.
import numpy as np
import matplotlib.pyplot as plt
x = np.arange(8)
y1 = 4 * x + 5
y2 = 3 * x + 5
y3 = 2 * x + 5
y4 = x + 5
colors = ["orange", "purple", "green", "red"]
plt.gca().set_prop_cycle(color=colors)
plt.plot(x, y1, label="4x+5")
plt.plot(x, y2, label="3x+5")
plt.plot(x, y3, label="2x+5")
plt.plot(x, y4, label="x+5")
plt.title("Plot Multiple lines in Matplotlib", fontsize=15)
plt.xlabel("X", fontsize=13)
plt.ylabel("Y", fontsize=13)
plt.legend()
plt.show()
출력:
첫 번째 줄은 주황색, 두 번째 줄은 보라색, 세 번째 줄은 녹색, 네 번째 줄은 빨간색으로 설정합니다. 순서대로 사용할 색상 목록을matplotlib.axes.Axes.set_prop_cycle()메서드에 인수로 전달합니다.
Suraj Joshi is a backend software engineer at Matrice.ai.
LinkedIn