included csv function

This commit is contained in:
cid
2023-12-23 23:55:11 +08:00
parent d0c541a318
commit 8b837b2d7c

View File

@ -21,7 +21,7 @@ def scrape_headings(url):
# Find all the heading elements (h1, h2, h3, etc.)
headings = soup.find_all(['h1', 'h2', 'h3', 'h4', 'h5', 'h6'])
# Extract and print the text from the headings and put to csv
# Extract and put to csv the result of headings
heading_list = [heading.text for heading in headings]
df = pd.DataFrame(heading_list)
df.to_csv('output.csv')