python中shape的用法

2024年2月3日发(作者：)

python中shape的用法

Python中shape的用法

在Python中，shape是一个在numpy库中的函数，用于获取数组的维度信息。它可以用于多种情况，下面将详细介绍一些常用的用法。

获取一维数组的长度

可以使用shape函数获取一维数组的长度。例如，考虑以下示例代码：

import numpy as np

arr = ([1, 2, 3, 4, 5])

arr_shape = (arr)

print(arr_shape)

输出结果为：

(5,)

在这里，shape函数返回了一个包含一个元素的元组(5,)，表示数组arr只有一个维度，并且长度为5。

获取多维数组的形状

shape函数也可以用于获取多维数组的形状信息。下面是一个示例代码：

import numpy as np

arr = ([[1, 2, 3], [4, 5, 6]])

arr_shape = (arr)

print(arr_shape)

输出结果为：

(2, 3)

在这个例子中，数组arr是一个2行3列的二维数组。shape函数返回了一个包含两个元素的元组(2, 3)，表示数组arr有两个维度，第一个维度的长度是2，第二个维度的长度是3。

用于其他数据结构的shape

除了多维数组外，shape函数还可以用于其他一些数据结构，例如字符串、列表等。下面是一个示例代码：

import numpy as np

string = "Hello, World!"

string_shape = (string)

print(string_shape)

alist = [1, 2, 3, 4, 5]

alist_shape = (alist)

print(alist_shape)

输出结果为：

(13,)

(5,)

在这个例子中，shape函数分别获取了字符串Hello, World!和列表[1, 2, 3, 4, 5]的形状信息。字符串的形状是(13,)，表示字符串的长度为13；而列表的形状是(5,)，表示列表只有一个维度，并且长度为5。

结论

shape函数是numpy库中一个有用的函数，可以用于获取数组和其他数据结构的形状信息。通过shape函数，我们可以轻松地获取数组的维度、长度等信息，从而方便地进行后续的处理和分析。在使用shape函数时，需要注意它返回的结果是一个元组类型，可以通过索引访问其中的元素来获取具体的形状信息。