Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Matplotlib을 사용하여 분산 추세선을 어떻게 그릴 수 있습니까?

<시간/>

matplotlib를 사용하여 산포 추세선을 그리려면 polyfit()을 사용할 수 있습니다. 및 poly1d() 추세선 포인트를 얻는 방법.

단계

  • Figure 크기를 설정하고 서브플롯 사이와 주변의 패딩을 조정합니다.

  • numpy를 사용하여 x 및 y 데이터 포인트를 만듭니다.

  • Figure와 서브플롯 세트를 생성합니다.

  • numpy를 사용하여 x 및 y 데이터 포인트를 플로팅합니다.

  • polyfit()을 사용하여 추세선 데이터 포인트 찾기 및 poly1d() 방법.

  • 플롯 x 및 p(x) plot()을 사용하는 데이터 포인트 방법.

  • 그림을 표시하려면 show()를 사용하세요. 방법.

import numpy as np
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.random.rand(100)
y = np.random.rand(100)
fig, ax = plt.subplots()
_ = ax.scatter(x, y, c=x, cmap='plasma')
z = np.polyfit(x, y, 1)
p = np.poly1d(z)
plt.plot(x, p(x), "r-o")

plt.show()

출력

Matplotlib을 사용하여 분산 추세선을 어떻게 그릴 수 있습니까?