Получить данные с помощью Beautifulsoup и сохранить их в словарь Python - PullRequest
Новая
       5

Получить данные с помощью Beautifulsoup и сохранить их в словарь Python

0 голосов
/ 03 июня 2018

Я пытаюсь получить HTML-файл, используя Beautifulsoap.Позже я хочу показать данные, создав REST API в формате JSON.REST API работает нормально.Но я столкнулся с проблемой структурирования данных в ожидаемом формате.Итак, я прилагаю код Python, который просто обрабатывает извлеченные данные.

HTML: - 

<!DOCTYPE html>
<html>
<head>
<style>

table, th, td {
    border: 1px solid black;
}
</style>
</head>

<body>

<table>
  <thead>
    <tr>
      <th>Date</th>
      <th>Savings</th>
      <th>Expenses</th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <td>January</td>
      <td>$100</td>
      <td>$200</td>
    </tr>
    <tr>
      <td>February</td>
      <td>$80</td>
      <td>$300</td>
    </tr>
  </tbody>
</table>

</body>
</html>


My expected output should be :- 

{
    "data":
        "Savings" : {
             "Janunary" : $100,
             "February" : $80
         },    
        "Expenses" : {
             "January" : $200,
             "February" : $300
        }
}

Код Python, который я написал, 

bs_content = BeautifulSoup(ra.body, 'html.parser') #this parse the whole html

headers = []
result = defaultdict(dict)

table = bs_content.find_all('table')

if not headers:
    for th in table.find('thead').findAll('th'):
        text = th.text
        headers.append(text)

for tr in table.find('tbody').findAll('tr'):
    tds = tr.findAll('td')
    for header, td in zip(headers, tds):
        value = td.text.strip()

        result[header] = value

return result

Таким образом, result должно быть обновлено как 

result['savings']['January'] = $100,
result['savings']['February'] = $80,
result['Expenses']['January'] = $200,
result['Expenses']['February'] = $300

1 Ответ

0 голосов
/ 04 июня 2018

Это решение должно работать для таблицы, которая имеет более 2 месяцев. 

soup = BeautifulSoup(ra.body, 'lxml')
table = soup.select_one('table')
headers = [header.text for header in table.select('th')][1:]
result = {headers[0]: {}, headers[1]: {}}
for row in table.select('tbody tr'):
    data = [value.text for value in row.select('td')]
    result[headers[0]][data[0]] = data[1]
    result[headers[1]][data[0]] = data[2]
return result
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.

Нет похожих вопросов

...