import numpy as np
print(np.arange(0, 3))
结果为:
[0 1 2]
如何改变它的维度呢?
(1) shape方法
import numpy as np
L1 = np.arange(0, 3)
L1.shape = (3, 1)
print(L1)
结果为:
[[0]
[1]
[2]]
(2) 使用np.newaxis
import numpy as np
print(np.arange(0, 3)[:, np.newaxis])
结果为:
[[0]
[1]
[2]]
根据np.newaxis
的前后位置不同,有:
import numpy as np
print(np.arange(0, 3)[np.newaxis, :])
结果为:
[[0 1 2]]