python怎么查看数据集的形状

在Python中，可以使用NumPy库来查看数据集的形状。以下是两种常用的方法：

1. 使用shape属性：如果数据集是一个NumPy数组，可以使用shape属性来查看其形状。示例代码如下：

import numpy as np

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

输出结果为(2, 3)，表示数据集有2行和3列。

2. 使用ndim和size属性：可以使用ndim属性获取数据集的维度数量，使用size属性获取数据集中元素的总数。示例代码如下：

import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6]])
print(data.ndim)  # 输出结果为2
print(data.size)  # 输出结果为6

这种方法适用于任何可迭代的数据集，不仅限于NumPy数组。