軸対称モデルを3次元化する

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12

 
 
 
-
!
 
 
 
 
 
 
 

from copy import deepcopy
import numpy as np
 
# Here, the vector array is assumed to be named velocity.
outputArray = deepcopy(velocity)
r = np.sqrt(points[:,0]**2 + points[:,1]**2)
ids = np.where(r>0)
 
outputArray[ids,0] = velocity[ids,0] * points[ids,0] / r[ids]
outputArray[ids,1] = velocity[ids,0] * points[ids,1] / r[ids]
 
return outputArray