numpy的dim函数如何使用

NumPy中的`dim()`函数用于返回给定数组的维度数量。具体用法如下：

import numpy as np

# 定义一个二维数组
arr = np.array([[1, 2], [3, 4], [5, 6]])

# 返回数组的维度数量
dim = np.ndim(arr)

print("数组arr的维度数量为：", dim)

输出结果为：

数组arr的维度数量为： 2

在上述示例中，我们首先导入NumPy库。然后，我们定义了一个二维数组`arr`，并使用`dim()`函数返回了该数组的维度数量。最后，我们将结果输出到控制台。