我知道如何对标题名称进行硬编码,但是我需要从数组中生成它们。这可能吗?
我的数据是动态抓取的,因此无法对标题或列进行硬编码
results_headings包含动物,矿物,蔬菜等字符串
results_columns包含诸如Bear,Quartz和Brocolli之类的字符串
我的密码
#Imports
from bs4 import BeautifulSoup
import requests
import pandas as pd
#Specify URL & Assign to page object
url = 'http://www.example.com'
page = requests.get(url)
#Grab our page as text
page.text
soup = BeautifulSoup(page.text, 'html.parser') #Use the HTML Parser
#Find our information
boxinfo = soup.find("div", {"id": "box1"})
headings = boxinfo.find_all("td", {"class": "label"})
columns = boxinfo.find_all("td")
#Get the headings
results_headings = []
for result in headings:
result_NoHTML = result.getText()
results.append(result_NoHTML)
#Get the columns
results_columns = []
for result2 in columns:
result2_NoHTML = result2.getText()
results_columns.append(result2_NoHTML)
df = pd.DataFrame(results_headings, results_columns)
df.to_csv('index.csv', index=False, encoding='utf-8')
我从中抓取的表格结构
<div class="box1">
<table class="table1">
<tr><td class="label">Item1</td><td>Value1</td></tr>
<tr><td class="label">Item2</td><td>Value2</td></tr>
<tr><td class="label">Item3</td><td>Value3</td></tr>
<tr><td class="label">Item4</td><td>Value4</td></tr>
</table>
</div>
最佳答案
因此,您已抓取数据并最终得到如下数据框。请注意,列仍未命名,但列名显示在第一行中,与数据没有任何分隔:
df = pd.DataFrame([['Animal', 'Mineral', 'Vegetable'],
['Bear', 'Quartz', 'Brocolli'],
['Turtle', 'Amethyst', 'Asparagus']])
print(df)
0 1 2
0 Animal Mineral Vegetable
1 Bear Quartz Brocolli
2 Turtle Amethyst Asparagus
您可以从第二行开始构造一个新的数据框,并将第一行分配为列:
df = pd.DataFrame(df.values[1:], columns=df.values[0])
print(df)
Animal Mineral Vegetable
0 Bear Quartz Brocolli
1 Turtle Amethyst Asparagus