pandas __iter__() 迭代操作

__iter__() 方法是 Python 中的一个特殊方法，用于返回一个对象的迭代器。在 pandas 中，DataFrame 和 Series 都实现了这个方法，使得它们可以被迭代。

支持的数据类型

• pandas.DataFrame
• pandas.Series

语法

__iter__() 方法没有参数，其语法如下：

DataFrame.__iter__()
Series.__iter__()

参数说明

__iter__() 方法不接受任何参数。

返回值

• DataFrame: 返回列名的迭代器。
• Series: 返回索引标签的迭代器。

使用场景

• 遍历列名：对于 DataFrame，可以通过 __iter__() 方法来遍历所有的列名。
• 遍历索引标签：对于 Series，可以通过 __iter__() 方法来遍历所有的索引标签。

例子

例1：遍历 DataFrame 的列名

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]
})

print("Initial DataFrame:")
print(df)

# 遍历列名
print("\nIterating over DataFrame columns:")
for col in df:
    print(col)

输出：

Initial DataFrame:
   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9

Iterating over DataFrame columns:
A
B
C

例2：遍历 Series 的索引标签

# 创建一个 Series
s = pd.Series([1, 2, 3], index=['a', 'b', 'c'])

print("Initial Series:")
print(s)

# 遍历索引标签
print("\nIterating over Series index:")
for idx in s:
    print(idx)

输出：

Initial Series:
a    1
b    2
c    3
dtype: int64

Iterating over Series index:
1
2
3

（注意：Series 的 __iter__() 方法返回的是其值的迭代器，而不是索引标签）

小结

__iter__() 方法是 Python 迭代协议的一部分，它使得 pandas 的 DataFrame 和 Series 对象可以被迭代。对于 DataFrame，迭代返回列名；对于 Series，迭代返回值。这种迭代功能可以用于遍历数据结构中的元素，便于数据的处理和分析。

参考

• https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.__iter__.html

更新时间：Aug. 11, 2024, 10:04 a.m. 标签：pandas python iter 迭代