避免使用numpy向量進行循環

Question

我正在創建許多不同角度的直線的x和y坐標。 如何將以下代碼矢量化，並避免使用for循環？

# set up vector line equation that goes through 180 deg
v_1 = np.array([0,0]) #step on vector
mu = np.linspace(0, 2.5, 1000)
angle_step = 100
theta = np.linspace(0, np.pi, angle_step) #180 deg range
v_2 = np.array([np.cos(theta), np.sin(theta)]) #gradient vector

for i in range(angle_step):
    # for every angle, generate x and y coordinates in +ve and -ve mu direction from centre
    x1, y1 = np.rint(v_1[0] + mu * v_2[:, i][0]).astype(int), np.rint(v_1[1] + mu * v_2[:, i][1]).astype(int)
    x2, y2 = np.rint(v_1[0] - mu * v_2[:, i][0]).astype(int), np.rint(v_1[1] - mu * v_2[:, i][1]).astype(int)

Answer 1

我認為這是您需要的：

x1, y1 = np.rint(v_1[0] + mu[:,None] * v_2[0]).astype(int), np.rint(v_1[1] + mu[:,None] * v_2[:, i][1]).astype(int)
x2, y2 = np.rint(v_1[0] - mu[:,None] * v_2[:, i][0]).astype(int), np.rint(v_1[1] - mu[:,None] * v_2[:, i][1]).astype(int)

mu[:,None]將mu更改為形狀數組(1000,1) ，可以通過v_2[0]進行廣播。