帮我写出用python爬取上市公司财务报表并存入excel的代码
时间: 2023-06-07 07:08:02 浏览: 605
好的,我能够帮您完成这个任务。以下是Python爬取上市公司财务报表并存入Excel的代码,您可以参考一下:
```python
import requests
from bs4 import BeautifulSoup
import pandas as pd
def get_finance_report(code):
url = 'http://quotes.money.163.com/f10/zycwzb_{}.html'.format(code)
response = requests.get(url)
response.encoding = 'utf-8'
soup = BeautifulSoup(response.text, 'html.parser')
report_table = soup.find('table', {'class': 'table_bg001 border_box limit_sale'})
report_data = []
headers = []
for index, tr in enumerate(report_table.find_all('tr')):
if index == 0:
headers = [th.text for th in tr.find_all('th')]
else:
row_data = []
for td in tr.find_all('td'):
row_data.append(td.text)
report_data.append(row_data)
return headers, report_data
def save_to_excel(headers, data, code):
df = pd.DataFrame(data, columns=headers)
writer = pd.ExcelWriter('{}财务报表.xlsx'.format(code))
df.to_excel(writer, index=False)
writer.save()
if __name__ == '__main__':
code = '600519'
headers, report_data = get_finance_report(code)
save_to_excel(headers, report_data, code)
```
其中,`code` 参数表示上市公司的股票代码。这个代码会构建出对应的股票财务报表的 URL 地址,并从网页中解析出表格数据。最后,将表格数据存入 Excel 文件中。注意,需要安装 `pandas` 和 `beautifulsoup4` 库。
阅读全文