python 如何读取数组长度

在Python中，有几种方法可以读取数组（即列表）的长度。你可以使用内置函数len()、使用numpy库中的shape属性、使用pandas库中的shape属性。其中，最常用和最简单的方法是使用len()函数。下面详细讲解如何使用len()函数来读取数组长度。

使用len()函数

len()函数是Python的内置函数，可以用于获取列表、字符串、字典等容器的长度。其语法非常简单，只需将数组作为参数传递给len()函数即可。下面是一个示例代码：

# 定义一个数组
my_list = [1, 2, 3, 4, 5]
使用len()函数获取数组长度
length = len(my_list)
打印数组长度
print(f"数组的长度是: {length}")

在这个示例中，我们定义了一个包含5个元素的数组my_list，然后使用len(my_list)来获取其长度，最后将长度打印出来。输出将是：

数组的长度是: 5

接下来，我们将详细介绍另外两种方法。

一、使用len()函数

1、基本用法

len()函数可以用于各种容器对象，包括列表、元组、字符串、字典等。其语法非常简单，直接将目标容器对象传递给len()函数即可。我们来看几个示例：

# 示例1：获取列表的长度
my_list = [1, 2, 3, 4, 5]
print(len(my_list))  # 输出: 5
示例2：获取字符串的长度
my_string = "Hello, World!"
print(len(my_string))  # 输出: 13
示例3：获取字典的长度
my_dict = {'a': 1, 'b': 2, 'c': 3}
print(len(my_dict))  # 输出: 3
示例4：获取元组的长度
my_tuple = (1, 2, 3)
print(len(my_tuple))  # 输出: 3

2、适用范围

len()函数适用于各种容器对象，但要注意的是，对于非容器对象（如整数、浮点数等），使用len()函数会导致TypeError。例如：

num = 123
print(len(num))  # 会引发TypeError: object of type 'int' has no len()

因此，在使用len()函数之前，确保目标对象是一个容器对象。

二、使用numpy库

1、基本用法

对于多维数组，numpy库提供了更强大的功能。numpy库中的ndarray对象有一个shape属性，可以返回数组的维度信息。我们来看一个示例：

import numpy as np
定义一个二维数组
my_array = np.array([[1, 2, 3], [4, 5, 6]])
获取数组的形状
shape = my_array.shape
打印数组的形状
print(f"数组的形状是: {shape}")

在这个示例中，我们定义了一个二维数组my_array，然后使用my_array.shape来获取其形状，最后将形状打印出来。输出将是：

数组的形状是: (2, 3)

2、获取数组的长度

对于一维数组，可以直接使用len()函数获取其长度；对于多维数组，可以通过shape属性的第一个元素来获取其长度。例如：

import numpy as np
定义一个一维数组
one_d_array = np.array([1, 2, 3, 4, 5])
print(len(one_d_array))  # 输出: 5
定义一个二维数组
two_d_array = np.array([[1, 2, 3], [4, 5, 6]])
print(two_d_array.shape[0])  # 输出: 2

三、使用pandas库

1、基本用法

pandas库主要用于数据分析，其核心数据结构是DataFrame和Series。DataFrame对象有一个shape属性，可以返回其形状信息。我们来看一个示例：

import pandas as pd
定义一个DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
获取DataFrame的形状
shape = df.shape
打印DataFrame的形状
print(f"DataFrame的形状是: {shape}")

在这个示例中，我们定义了一个DataFrame对象df，然后使用df.shape来获取其形状，最后将形状打印出来。输出将是：

DataFrame的形状是: (3, 2)

2、获取数据长度

对于Series对象，可以直接使用len()函数获取其长度；对于DataFrame对象，可以通过shape属性的第一个元素来获取其行数。例如：

import pandas as pd
定义一个Series
series = pd.Series([1, 2, 3, 4, 5])
print(len(series))  # 输出: 5
定义一个DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
print(df.shape[0])  # 输出: 3

四、其他方法

1、使用collections模块

collections模块提供了许多有用的数据结构，如deque、Counter等。这些数据结构也可以使用len()函数来获取其长度。例如：

from collections import deque, Counter
定义一个deque
d = deque([1, 2, 3, 4, 5])
print(len(d))  # 输出: 5
定义一个Counter
c = Counter({'a': 1, 'b': 2, 'c': 3})
print(len(c))  # 输出: 3

2、自定义数据结构

如果你定义了自己的数据结构，可以通过实现__len__方法来使其支持len()函数。例如：

class MyList:
    def __init__(self, elements):
        self.elements = elements
    def __len__(self):
        return len(self.elements)
定义一个MyList对象
my_list = MyList([1, 2, 3, 4, 5])
print(len(my_list))  # 输出: 5

在这个示例中，我们定义了一个自定义数据结构MyList，并实现了__len__方法，使其支持len()函数。这样，我们就可以像使用内置列表一样使用len()函数来获取其长度。