Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python Matplotlib 3D 산점도에서 두 점 연결하는 방법

3D 산점도(scatter plot)에서 두 점을 연결하려면 다음 단계를 따르면 됩니다.

  • 그림(figure) 크기를 설정하고 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.
  • figure() 메서드를 사용하여 새 그림을 생성하거나 기존 그림을 활성화합니다.
  • subplot 배치 방식으로 현재 그림에 3D 축(axes)을 추가합니다.
  • x, y, z 좌표 데이터를 리스트로 생성합니다.
  • scatter() 메서드를 사용하여 x, y, z 데이터 포인트를 산점도로 표시합니다.
  • 두 점을 연결하려면 plot() 메서드에 x, y, z 데이터를 전달하고 검은색(black) 선으로 지정합니다.
  • show() 메서드를 호출하여 완성된 그림을 화면에 표시합니다.

예제 코드

from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
fig = plt.figure()
ax = fig.add_subplot(projection="3d")
x, y, z = [1, 1.5], [1, 2.4], [3.4, 1.4]
ax.scatter(x, y, z, c='red', s=100)
ax.plot(x, y, z, color='black')
plt.show()

실행 결과

위 코드를 실행하면 빨간색 점 두 개가 3D 공간에 표시되고, 두 점 사이는 검은색 직선으로 연결된 것을 확인할 수 있습니다. scatter() 메서드의 c 매개변수로 점의 색상을, s 매개변수로 점의 크기를 조절할 수 있으며, plot() 메서드의 color 매개변수로 연결선의 색상을 자유롭게 변경할 수 있습니다.