pandas filter 使用
时间: 2023-07-11 22:37:18 浏览: 114
pandas的`filter()`函数用于从DataFrame中过滤出符合指定条件的列,常用于选择特定的列进行操作。该函数的语法如下:
```
DataFrame.filter(items=None, like=None, regex=None, axis=None)
```
其中,各参数的含义如下:
- `items`:要保留的列名列表。
- `like`:列名中包含指定字符串的列。
- `regex`:列名符合正则表达式的列。
- `axis`:要过滤的轴,默认为列(axis=1)。
下面是一个简单的示例,假设有一个DataFrame df,其中包含三列'A'、'B'、'C':
```
import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)
print(df)
```
输出结果为:
```
A B C
0 1 4 7
1 2 5 8
2 3 6 9
```
现在我们想要保留列'A'和列'C',可以使用如下代码:
```
df.filter(items=['A', 'C'])
```
输出结果为:
```
A C
0 1 7
1 2 8
2 3 9
```
如果我们想要选择列名中包含字母'B'的列,可以使用如下代码:
```
df.filter(like='B')
```
输出结果为:
```
B
0 4
1 5
2 6
```
如果我们想要选择列名以字母'C'结尾的列,可以使用如下代码:
```
df.filter(regex='C$')
```
输出结果为:
```
C
0 7
1 8
2 9
```
阅读全文