numpy.newaxis的作用和示例

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]]]