首页 > Python教程 > Python NumPy库 阅读数:4

NumPy ndarray的常用属性

ndarray 是一个类,其包含一些属性。

最基本的便是其维度。可以用属性 ndim 来得到指定矩阵的维度,方法如下:
>>> a = np.array([1, 2, 3])
>>> a.ndim                            # 维度为1
1
>>> b = np.eye(3)
>>> b.ndim                            # 维度为2
2

可以用属性 shape 来得到指定数组的外形,方法如下:
>>> a = np.eye(3)                    # 3x3的单元矩阵
>>> a                                # 查看a的值
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])
>>> a.shape                            # a的外形
(3, 3)

可以用属性 dtype 来得到指定矩阵每个元素的类型,方法如下:
>>> a = np.eye(3, dtype=int)        # 指定类型为整型
>>> a
array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 1]])
>>> a.dtype                            # 查看类型
dtype('int64')

可以用属性 size 来得到指定矩阵的元素个数,方法如下:
>>> a = np.eye(3, dtype=int)
>>> a
array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 1]])
>>> a.size                                    # 矩阵元素个数,9个
9

可以用属性 T 来得到指定矩阵的转置矩阵,方法如下:
>>> a = np.array([1, 2, 3])                    # 一维矩阵
>>> a.T                                        # 转置矩阵是自己
array([1, 2, 3])
>>> b = np.array([[1, 2, 3], [4, 5, 6]])    # 二维矩阵
>>> b
array([[1, 2, 3],
       [4, 5, 6]])
>>> b.T                                        # 转置矩阵
array([[1, 4],
       [2, 5],
       [3, 6]])

需要注意的是,转置矩阵就是将原矩阵旋转 90 度得到的矩阵,但其仅对二维以及多维矩阵有效。对于一维矩阵来说,其转置矩阵还是自己。