import pandas as pd keyword = ['涪陵','长寿','城二'] def fl_get(x): # 读取原始表格数据 df = pd.read_excel('date2.0.xlsx') # 提取包含涪陵字段的数据行 filtered_df = df[df['分公司'].str.contains(x)] # 保存到另一个Excel文件 filtered_df.to_excel('%s'%(x)+'.xlsx', index=False) fl_get() for x in range(keyword): fl_get(x)
时间: 2023-09-15 17:20:44 浏览: 173
python数据分析与可视化 import pandas as pd import numpy as np import m
在你的代码中,有几点需要进行修改:
1. 在函数`fl_get()`中,你使用了一个参数`x`作为关键字进行筛选,但是在调用函数的时候没有传入任何参数。你可以将`fl_get()`函数的定义修改为`fl_get(x)`,然后在调用函数时传入关键字。
2. 在循环中,你使用了`range(keyword)`来遍历关键字列表,但是`range()`函数需要一个整数作为参数,而不是一个列表。你可以使用`for x in keyword`来遍历关键字列表。
3. 在保存Excel文件时,你使用了关键字作为文件名,但是没有添加文件后缀名。你可以在文件名中添加`.xlsx`后缀。
下面是修改后的代码:
```python
import pandas as pd
keyword = ['涪陵', '长寿', '城二']
def fl_get(x):
# 读取原始表格数据
df = pd.read_excel('date2.0.xlsx')
# 提取包含关键字字段的数据行
filtered_df = df[df['分公司'].str.contains(x)]
# 保存到另一个Excel文件
filtered_df.to_excel('%s.xlsx' % x, index=False)
for x in keyword:
fl_get(x)
```
这样,代码就可以按照关键字提取并保存对应的数据行到不同的Excel文件中了。记得根据实际情况修改文件名和列名,并确保已经安装了pandas库。
阅读全文