在Python中,可以使用NumPy库来查看数据集的形状。以下是两种常用的方法:
shape
属性:如果数据集是一个NumPy数组,可以使用shape
属性来查看其形状。示例代码如下:import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6]])
print(data.shape)
输出结果为(2, 3)
,表示数据集有2行和3列。
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数组。