Numpy的size，shape和ndim属性

Numpy的size，shape和ndim属性

（新手上路）
设a为一个数组

1.size属性
a.size展示的是数组a的规模，就是a的元素个数为多少。

import numpy as np

def test() :
    true = np.array([1, 2])
    pred = np.array([[2, 4], [8, 16]])
    print(true.size, pred.size)
 
test()

运行结果

2.shape属性
a.shape就是数组a在各个维度的大小，比如（2， 3， 4）代表a的大小是2 * 3 * 4。

import numpy as np

def test() :
    true = np.array([1, 2])
    pred = np.array([[2, 4], [8, 16]])
    print(true.shape, pred.shape)

test()

3.ndim属性
a.ndim就是数组a有几个维度，比如3就代表数组a有三个维度

import numpy as np

def test() :
    true = np.array([1, 2])
    pred = np.array([[2, 4], [8, 16]])
    print(true.ndim, pred.ndim)

test()