numpy.newaxis的作用和示例
释义:在 numpy.newaxis 所在的位置增加一维
示例1:
import numpy as np
X = np.array([1, 2, 3])
print(X)
print('='*30)
print(X[:, np.newaxis])
[1 2 3]
==============================
[[1]
[2]
[3]]
print(X[np.newaxis, :])
[[1 2 3]]
示例2:
X = np.array([[1, 2, 3],
[4, 5, 6]])
print(X)
print('='*30)
print(X[:, :, np.newaxis])
[[1 2 3]
[4 5 6]]
==============================
[[[1]
[2]
[3]]
[[4]
[5]
[6]]]
print(X[:, np.newaxis, :])
[[[1 2 3]]
[[4 5 6]]]
print(X[np.newaxis, :, :])
[[[1 2 3]
[4 5 6]]]