在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數組。