温馨提示×

python怎么查看数据集的形状

小亿
301
2023-12-21 18:00:05
栏目: 编程语言

在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列。

  1. 使用ndimsize属性:可以使用ndim属性获取数据集的维度数量,使用size属性获取数据集中元素的总数。示例代码如下:
import numpy as np

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

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

0