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

Python에서 벡터 필드의 컬을 계산하고 Matplotlib로 플로팅합니다.

<시간/>

Python에서 벡터 필드의 컬을 계산하고 Matplotlib로 플롯하려면 quiver()를 사용할 수 있습니다. 방법 및 해당 데이터를 계산합니다.

단계

  • 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
  • Figure()를 사용하여 새 Figure 생성 또는 기존 Figure 활성화 방법.
  • 서브플롯 배열의 일부로 Figure에 3D 축을 추가합니다.
  • x 만들기 , yz numpy meshgrid를 사용한 데이터 포인트
  • u 만들기 , v 그리고 w 데이터 컬 벡터 위치.
  • quiver() 사용 벡터를 가져오는 방법입니다.
  • 축을 끕니다.
  • 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import matplotlib.pyplot as plt
import numpy as np

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 = np.meshgrid(np.arange(-0.8, 1, 0.2),
                     np.arange(-0.8, 1, 0.2),
                     np.arange(-0.8, 1, 0.8))

u = 0
v = y**2
w = -2*y*z - y

ax.quiver(x, y, z, u, v, w, length=0.1)
ax.axis('off')

plt.show()

출력

Python에서 벡터 필드의 컬을 계산하고 Matplotlib로 플로팅합니다.