numpy.newaxis 转变矩阵的形狀   2018-09-11


有一個一維陣列 x1,我分別想要把它變成一個 3*1 的矩陣 x2,以及 1*3 的矩陣 x3,作法如下。

import numpy as np

x1 = np.array([10, 20, 30], float)

# 有一個一維陣列x1,我分別想要把它變成一個 3*1 的矩陣x2,以及 1*3 的矩陣x3,作法如下。

print("shape of x1 is ", x1.shape)
print(x1)

print("-------------")

x2 = x1[:, np.newaxis]
print("shape of x2 is ", x2.shape)
print(x2)

print("-------------")

x3 = x1[np.newaxis, :]
print("shape of x3 is ", x3.shape)
print(x3)

output:

shape of x1 is  (3,)
[ 10. 20. 30.]
-------------
shape of x2 is (3, 1)
[[ 10.]
[ 20.]
[ 30.]]
-------------
shape of x3 is (1, 3)
[[ 10. 20. 30.]]

Reference


分享到:


  如果您觉得这篇文章对您的学习很有帮助, 请您也分享它, 让它能再次帮助到更多的需要学习的人. 您的支持将鼓励我继续创作 !
本文基于署名4.0国际许可协议发布,转载请保留本文署名和文章链接。 如您有任何授权方面的协商,请邮件联系我。

Contents

  1. Reference