Normalize vector with numpy

README.md

So you want to normalize a vector? Well numpy can do that for you with a little help.

First include numpy:

!INCLUDE normalized.py 1:1

Then call the norm function of the linealg module:

!INCLUDE normalized.py 4:6

Encapsulate all this into a function:

!INCLUDE normalized.py

And now with graphics!

@startditaa
+--------+   +-------+    +-------+
|        +---+ ditaa +--> |       |
|  Text  |   +-------+    |diagram|
|Document|   |!magic!|    |       |
|     {d}|   |       |    |       |
+---+----+   +-------+    +-------+
	:                         ^
	|       Lots of work      |
	+-------------------------+
@endditaa

That's all folks!

meta.yml

title: Normalize a vector with numpy

normalized.py

import numpy as np

def normalized(a, axis=-1, order=2):
  l2 = np.atleast_1d(np.linalg.norm(a, order, axis))
  l2[l2==0] = 1
  return (a / np.expand_dims(l2, axis))[0]