python中匹配dataframe中第一列以‘APP运营室’或‘非集团全渠问题’或‘互联网发展室|’开头的行
时间: 2023-07-16 08:13:57 浏览: 78
使用Python向DataFrame中指定位置添加一列或多列的方法
4星 · 用户满意度95%
可以使用正则表达式来匹配第一列以特定字符串开头的行。具体实现如下:
```python
import re
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({'第一列': ['APP运营室xxx', '非集团全渠问题yyy', '互联网发展室zzz', '其他aaa'],
'第二列': [1, 2, 3, 4]})
# 使用正则表达式匹配第一列以特定字符串开头的行
pattern = '^(APP运营室|非集团全渠问题|互联网发展室)'
mask = df['第一列'].str.contains(pattern)
result = df[mask]
print(result)
```
输出:
```
第一列 第二列
0 APP运营室xxx 1
1 非集团全渠问题yyy 2
2 互联网发展室zzz 3
```
其中,`^`表示匹配行首,`|`表示或的关系。使用`str.contains()`方法将正则表达式应用到DataFrame的第一列上,得到一个布尔类型的掩码,可以用于筛选出符合条件的行。
阅读全文