Skip to content

Commit

Permalink
pep8 lint fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
perimosocordiae committed Oct 29, 2014
1 parent 1a7e2fb commit b8dccc0
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions lib/mpl_toolkits/mplot3d/axes3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -2569,15 +2569,15 @@ def calc_arrow(uvw, angle=15):

if len(XYZ) > 0:
# compute the shaft lines all at once with an outer product
shafts = (XYZ - np.multiply.outer(shaft_dt, UVW)).swapaxes(0,1)
shafts = (XYZ - np.multiply.outer(shaft_dt, UVW)).swapaxes(0, 1)
# compute head direction vectors, n heads by 2 sides by 3 dimensions
head_dirs = np.array([calc_arrow(d) for d in UVW])
# compute all head lines at once, starting from where the shaft ends
heads = shafts[:,:1] - np.multiply.outer(arrow_dt, head_dirs)
heads = shafts[:, :1] - np.multiply.outer(arrow_dt, head_dirs)
# stack left and right head lines together
heads.shape = (len(arrow_dt), -1, 3)
# transpose to get a list of lines
heads = heads.swapaxes(0,1)
heads = heads.swapaxes(0, 1)

lines = list(shafts) + list(heads)
else:
Expand All @@ -2586,7 +2586,7 @@ def calc_arrow(uvw, angle=15):
linec = art3d.Line3DCollection(lines, *args[argi:], **kwargs)
self.add_collection(linec)

self.auto_scale_xyz(XYZ[:,0], XYZ[:,1], XYZ[:,2], had_data)
self.auto_scale_xyz(XYZ[:, 0], XYZ[:, 1], XYZ[:, 2], had_data)

return linec

Expand Down

0 comments on commit b8dccc0

Please sign in to comment.