如何在 NumPy 中获取向量的大小？

在 NumPy 中，获取向量大小可以使用 shape 属性或者 size 属性。

使用 shape 属性

向量的大小可以通过 shape 属性获取。shape 属性返回一个表示数组形状的元组，如果数组是一维的，则元组只有一个元素，表示向量大小。

import numpy as np

v = np.array([1, 2, 3, 4, 5])
size = v.shape[0]

print("向量 v 的大小是：", size)

上述代码输出结果为：

向量 v 的大小是： 5

使用 size 属性

向量的大小也可以通过 size 属性获取。size 属性返回数组中的元素个数，对于向量来说，它的大小就是元素个数。

import numpy as np

v = np.array([1, 2, 3, 4, 5])
size = v.size

print("向量 v 的大小是：", size)

上述代码输出结果为：

向量 v 的大小是： 5

可以看到，使用 shape 属性和 size 属性都可以获取向量的大小。但是，如果要获取数组的维度数量，应该使用 ndim 属性。