numpy中shape的部分解释

转载自：https://blog.csdn.net/qq_28618765/article/details/78081959和https://www.jianshu.com/p/e083512e4f4c

shape函数是numpy.core.fromnumeric中的函数，它的功能是读取矩阵的长度，比如shape[0]就是读取矩阵第一维度的长度。

shape的输入参数可以是一个整数（表示维度），也可以是一个矩阵。

• 参数是一个数时小括号内没有矩阵符号中括号[]，返回空：

>>> import numpy as np
>>> np.shape(0)
()

• 参数是一维矩阵(一个[])：

>>> import numpy as np
>>> np.shape([1])
(1,)
>>> np.shape([1, 2])
(2,)

• 参数是二维矩阵(两个[])：

>>> import numpy as np
>>> np.shape([[1],[2]])
(2, 1)
>>> np.shape([[1, 2], [2, 3], [3, 4]])
(3, 2)

• 直接用.shape可以快速读取矩阵的形状，使用shape[0]读取矩阵第一维度的长度

>>> import numpy as np
>>> a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> a
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
>>> a.shape
(3, 3)
>>> a.shape[0]
3
>>> a.shape[1]
3

• 但是当某一维度长度不一致时，读取所有维度时则不能读出长短不一致的维度

>>> import numpy as np
>>> a = np.array([[1, 2, 3], [4, 5], ])
>>> a.shape
>>> a.shape[0]
2
>>> a.shape[1]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
(2,)

链接：https://www.jianshu.com/p/e083512e4f4c