我是Python的初学者,我必须使用Numpy处理项目。 我需要在圆柱体表面的一部分上生成一些点(例如一百万个)。这些点应定期分布在由给定角度定义的表面的子区域上。我怎么能这样做?
我的输入参数是:
position
(例如[0,0,0]
)
圆柱的orientation
length
圆柱
radius
圆柱
angle
(这定义了应在其上分配点的圆柱部分。)alpha = 360
,整个表面
delta_l
是长度方向上每两个点之间的距离
delta_alpha
是alpha
(旋转)方向上每两个点之间的距离
我的输出参数:
有人可以帮助我,或者给我一些关于如何做到的提示吗?
非常感谢
答案 0 :(得分:3)
这取自我以前的一个项目:
def make_cylinder(radius, length, nlength, alpha, nalpha, center, orientation):
#Create the length array
I = np.linspace(0, length, nlength)
#Create alpha array avoid duplication of endpoints
#Conditional should be changed to meet your requirements
if int(alpha) == 360:
A = np.linspace(0, alpha, num=nalpha, endpoint=False)/180*np.pi
else:
A = np.linspace(0, alpha, num=nalpha)/180*np.pi
#Calculate X and Y
X = radius * np.cos(A)
Y = radius * np.sin(A)
#Tile/repeat indices so all unique pairs are present
pz = np.tile(I, nalpha)
px = np.repeat(X, nlength)
py = np.repeat(Y, nlength)
points = np.vstack(( pz, px, py )).T
#Shift to center
shift = np.array(center) - np.mean(points, axis=0)
points += shift
#Orient tube to new vector
#Grabbed from an old unutbu answer
def rotation_matrix(axis,theta):
a = np.cos(theta/2)
b,c,d = -axis*np.sin(theta/2)
return np.array([[a*a+b*b-c*c-d*d, 2*(b*c-a*d), 2*(b*d+a*c)],
[2*(b*c+a*d), a*a+c*c-b*b-d*d, 2*(c*d-a*b)],
[2*(b*d-a*c), 2*(c*d+a*b), a*a+d*d-b*b-c*c]])
ovec = orientation / np.linalg.norm(orientation)
cylvec = np.array([1,0,0])
if np.allclose(cylvec, ovec):
return points
#Get orthogonal axis and rotation
oaxis = np.cross(ovec, cylvec)
rot = np.arccos(np.dot(ovec, cylvec))
R = rotation_matrix(oaxis, rot)
return points.dot(R)
绘制点:
points = make_cylinder(3, 5, 5, 360, 10, [0,2,0], [1,0,0])
旋转部件快速且脏 - 您应该仔细检查它。 Euler-Rodrigues公式归功于unutbu。