Python numpy array massaging
2 answers
Add a new axis with None/np.newaxis
-
a[:,None,:] # Or simply a[:,None]
Example run -
In [222]: a = np.random.randint(0,9,(4,3))
In [223]: a
Out[223]:
array([[1, 6, 6],
[4, 4, 5],
[7, 4, 4],
[4, 1, 3]])
In [224]: a[:,None]
Out[224]:
array([[[1, 6, 6]],
[[4, 4, 5]],
[[7, 4, 4]],
[[4, 1, 3]]])
+4
source to share