帮我优化以上的代码,并将结果储存在excel里
时间: 2024-04-01 10:31:45 浏览: 13
好的,以下是一个优化后的Python爬虫示例,用于从巨潮资讯站(http://www.cninfo.com.cn/)上爬取上市公司的年度财务数据,并将结果储存在Excel文件中:
```python
import requests
from bs4 import BeautifulSoup
import openpyxl
# 待爬取的公司代码
code = '600519'
# 构造 URL
url = 'http://www.cninfo.com.cn/new/data/szse_stock{}.html'.format(code)
# 发送请求
response = requests.get(url)
# 解析 HTML
soup = BeautifulSoup(response.content, 'html.parser')
# 获取表格数据
table = soup.find_all('table')[0]
rows = table.find_all('tr')
# 将数据存储到Excel文件中
workbook = openpyxl.Workbook()
worksheet = workbook.active
for row in rows:
cols = row.find_all('td')
cols = [col.text.strip() for col in cols]
worksheet.append(cols)
# 保存Excel文件
workbook.save('financial_data.xlsx')
```
这个示例代码使用了openpyxl库来操作Excel文件,将获取的表格数据逐行添加到工作表中,最后通过调用save()方法将工作表保存到Excel文件中。你可以根据需要修改代码,例如添加异常处理、设置自定义输出格式等。