This line is overly complicated. Try this:
```
arrow_vectors = zip(*(calc_arrow(u, v, w) for u, v, w in zip(us, vs, ws)))
```
Or, maybe it makes sense to vectorize `calc_arrow`?