fig = plt.figure() ax = fig.add_subplot(111,projection='3d') pi = np.pi seg = Segment() seg.bend_angle = pi/2 seg.bend_direction = 3*pi rotvec = [0,0,1] seg.apply_rotvec(rotvec,[1,0,0]) x = seg.bend_line(rotvec) ax.plot(x[0],x[1],x[2],color='black') c = magpy.Collection(seg.magnet) magpy.displaySystem(c,subplotAx=ax,suppress=True) ax.set_xlabel('X') ax.set_ylabel('Y') ax.set_zlabel('Z') ''' ax.set_xlim(-200,200) ax.set_ylim(-200,200) ax.set_zlim(-200,200) ''' plt.show()