python用shape函数

在 Python 中，shape 函数是用于获取数组或矩阵的维度的函数。它可以用于 numpy 库中的数组对象和 pandas 库中的数据框对象。下面是使用 shape 函数的示例：

对于 numpy 数组：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
shape = arr.shape
print(shape)  # 输出 (2, 3)

对于 pandas 数据框：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
shape = df.shape
print(shape)  # 输出 (3, 2)

以上示例分别展示了如何使用 shape 函数获取 numpy 数组和 pandas 数据框的维度信息。