在Python中,shape
是一个用于获取数组或矩阵的形状的函数。在本攻略中,我们将详细讲解如何使用shape
函数,并提供两个示例说明。
使用shape
函数
在Python中,shape
函数是一个用于获取数组或矩阵的形状的函数。该函数返回一个元组,其中包含数组或矩阵的维度信息。以下是一个示例,演示了如何使用shape
函数:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape) # 输出:(2, 3)
在上面的示例中,我们首先导入了numpy
库,并创建了一个包含两个列表的数组。接下来,我们使用shape
函数获取数组的形状,并将其输出到控制台上。
使用reshape
函数
我们还可以使用reshape
函数来改变数组的形状,并使用shape
函数来验证形状是否已更改。以下是一个示例,示了如何使用reshape
函数:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6])
new_arr = arr.reshape(2, 3)
print(new_arr) # 输出:[[1 2 3]
# [4 5 6]]
print(new_arr.shape) # 输出:(2, 3)
在上面的示例中,我们首先创建了一个包含六个元素的数组。接下来,我们使用reshape
函数将数组的形状更改为2行3列的矩阵,并将其赋值给new_arr
变量。最后,我们使用print
函数输出new_arr
和new_arr
的形状。
结论
在Python中,shape
函数是一个用于获取数组或矩阵的形状的函数。该函数返回一个元组,其中包含数组或矩阵的维度信息。我们还可以使用reshape
函数来改变数组的形状,并使用shape
函数来验证形状是否已更改。这些函数对于处理数组和矩阵非常有用。