Python/Pandas Dataframe:自动填充缺少的行

2024-10-04 01:31:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我的目标是最终在x轴上创建一个散点图,并在y轴上赢得(每个候选人的)代表。我不知道该如何“填空”当提到错过的日期。我附上了一张我得到的桌子的照片。在

例如,我试图将3月1日作为阿拉斯加、阿肯色州等地的日期,以便绘制数据。在

# CREATE DATAFRAME WITH DELEGATE WON/TARGET INFORMATION

import requests 
from lxml import html 
import pandas 

url = "http://projects.fivethirtyeight.com/election-2016/delegate-targets/"
response = requests.get(url)
doc = html.fromstring(response.text)

tables = doc.findall('.//table[@class="delegates desktop"]')
election = tables[0] 
election_rows = election.findall('.//tr')
def extractCells(row, isHeader=False):
    if isHeader:
        cells = row.findall('.//th')
    else:
        cells = row.findall('.//td')
    return [val.text_content() for val in cells]


def parse_options_data(table):

    rows = table.findall(".//tr")
    header = extractCells(rows[1], isHeader=True)
    data = [extractCells(row, isHeader=False) for row in rows[2:]]

    trumpdata = "Trump Won Delegates"
    cruzdata = "Cruz Won Delegates"
    kasichdata = "Kasich Won Delegates"

    data = pandas.DataFrame(data, columns=["Date", "State or Territory", "Total Delegates", trumpdata, cruzdata, kasichdata, "Rubio"])

    data.insert(4, "Trump Target Delegates", data[trumpdata].str.extract(r'(\d{0,3}$)'))
    data.insert(6, "Cruz Target Delegates", data[cruzdata].str.extract(r'(\d{0,3}$)'))
    data.insert(8, "Kasich Target Delegates", data[kasichdata].str.extract(r'(\d{0,3}$)'))

    data = data.drop('Rubio', 1)
    data[trumpdata] = data[trumpdata].str.extract(r'(^\d{0,3})')
    data[cruzdata] = data[cruzdata].str.extract(r'(^\d{0,3})')
    data[kasichdata] = data[kasichdata].str.extract(r'(^\d{0,3})')

    return df

election_data = parse_options_data(election)
df = pandas.DataFrame(election_data)
df

Picture of my table


Tags: importpandasdatatableextractrowsrowstr