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

Matplotlib에서 scatter()를 사용하여 3D 산점도에 범례 추가

<시간/>

scatter()를 사용하여 3D 산점도에 범례를 추가하려면 matplotlib에서 다음 단계를 수행할 수 있습니다. -

단계

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

  • 변수 초기화 N 샘플 데이터의 수를 저장합니다.

  • x 만들기 및 y 데이터 포인트; z1을 만들다 및 z2 데이터 포인트 목록.

  • projection='3d'를 사용하여 현재 그림에 서브플롯을 추가합니다. .

  • x 플롯 , yz1 plot()을 사용하는 데이터 포인트 마커 *가 있는 2D 축의 점 .

  • x 플롯 , yz2 plot()을 사용하는 데이터 포인트 마커 o가 있는 2D 축의 점 .

  • 그림에 범례를 배치합니다.

  • 그림을 표시하려면 show()를 사용할 수 있습니다. 방법.

예시

import matplotlib.pyplot as plt
import numpy as np

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

N = 100

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

z1 = [i*j for (i, j) in zip(x, y)]
z2 = [i+j for (i, j) in zip(x, y)]

axes = plt.subplot(111, projection='3d')

axes.plot(x, y, z1, "*", label="xy")
axes.plot(x, y, z2, "o", label="x+y")

plt.legend(loc="upper right")

plt.show()

출력

다음 출력을 생성합니다 -

Matplotlib에서 scatter()를 사용하여 3D 산점도에 범례 추가