filter()
函数用于根据某些条件筛选 DataFrame 中的列或行。它接受一个参数 items
,用于指定要筛选的列名或行标签,还可以传递一个参数 like
、regex
或 axis
进行更加灵活的筛选。
下面是一个简单的示例:
import pandas as pd
# 创建一个 DataFrame
data = {
'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9],
'D': [10, 11, 12]
}
df = pd.DataFrame(data)
# 使用 filter() 函数筛选列名以 'A' 开头的列
result = df.filter(like='A')
print("Filtered columns:")
print(result)
Filtered columns:
A
0 1
1 2
2 3
在这个示例中,我们创建了一个 DataFrame,并使用 filter()
函数筛选了列名以 'A' 开头的列。你也可以根据其他条件筛选列或行,如根据正则表达式匹配、指定行标签等。