如何得到矩阵的大小？

Question

最近，我有一个项目需要知道矩阵的大小。 这是一段matlab代码。在

[rwn1 rwn2]=size(rtt)

然后您将知道它在名为“rtt”的矩阵中有多少行和列但是代码我想把matlab代码转换成python，我还创建了一个这样的python代码

^{pr2}$

它只打印了（10，） 它不打印行数，只打印列数。 我想要的结果是（10，1），有没有高手能帮我解决这个问题？在

Answer 1

您可以更改形状，但只需确保行/列顺序符合您的要求：

>>> rtt = np.array([1,2,3,4,5,6,7,8,9,10])
>>> rtt
array([ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
>>> rtt.shape
(10,)

你说你想要(10,1)：

^{pr2}$

但也许(1,10)就是你想要的：

>>> rtt.shape = (1,10)
>>> rtt
array([[ 1,  2,  3,  4,  5,  6,  7,  8,  9, 10]])

如果您想(2,5)，请自便：

>>> rtt.shape = (2,5)
>>> rtt
array([[ 1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10]])
>>> 

Answer 2

您将创建一个一维阵列。在

import numpy as np  
rtt=np.array([[1,2,3,4,5,6,7,8,9,10]]) 
#rtt=np.array([1,2,3,4,5,6,7,8,9,10],ndmin=2) another way
rtt.shape #(1, 10)

如果你想用（列，行）格式

^{pr2}$

Answer 3

numpy和matlab之间有一些区别，这可能会让新用户感到恼火。其中一个是这个，你需要这样做

    rtt = rtt.reshape(1,-1)

得到一个行矩阵。我想您会发现此文档页很有帮助：

https://docs.scipy.org/doc/numpy-dev/user/numpy-for-matlab-users.html