温馨提示×

怎么用numpy查看数据维度

小亿
205
2023-12-16 09:18:32
栏目: 编程语言

要使用 NumPy 查看数据的维度,你可以使用 `shape` 属性。以下是一个示例:

import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6]])
print(data.shape)

输出将显示数据的维度,例如 `(2, 3)`。这表示数据是一个二维数组,其中有两个行和三个列。

如果你有一个一维数组,你可以通过将其转换为二维数组来查看维度。以下是一个示例:

import numpy as np
data = np.array([1, 2, 3, 4, 5])
print(data.shape)

输出将显示 `(5,)`,表示该数组有五个元素,但只有一个维度。

另外,你还可以使用 `ndim` 属性来获取数据的维数。例如:

import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6]])
print(data.ndim)

输出将显示 `2`,表示数据是一个二维数组。对于一维数组, `ndim` 的输出将是 `1`。

0