python:如何改进代码片段以解析ord中的元素

2024-04-26 19:10:54 发布

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

我希望在html块中找到我希望找到的所有元素,然后对其进行排序,以便所有标记在我的列表中都有位置,例如:('h3'、'a'、'img')

我想知道是否有更好的方法来解决我的问题,在一个更漂亮的方式和更容易扩展(添加更多的标签)。你知道吗

例:这样我就可以把这个列表发送给一个函数,而不用三思而后行。你知道吗

以下是运行代码段后的结果:

[
('text 1', '/url1', '/img1.png'),
('text 2', '/url2', '/img3.png'),
('text 3', '/url3', '/img3.png'),
]

代码段:

def parse_element_tag(el):
    #<class 'lxml.html.HtmlElement'>
    dict = {'a': (el.get('href'), 1), 'img': (el.get('src'), 2), 'h3': (el.text, 0)}
    return dict[el.tag]

requests_cache.configure('test', expire_after=900)
r = readUrl('http://www.svtplay.se/program')
l = lxml.html.fromstring(r.text)
lst = []
for el in l.cssselect('div ul.svtGridBlock li div a'):
    #lst.append(sorted([parse_element_tag(i) for i in el.iter()], key=lambda val: val[1]))
    lst.append(
               tuple([i[0] for i in sorted(
                      [parse_element_tag(i) for i in el.iter() if i.tag in ('a', 'img', 'h3')], key=lambda val: val[1]
                      )]
               ))

Tags: textin列表imgforpngparsehtml
1条回答
网友
1楼 · 发布于 2024-04-26 19:10:54

使用xpath

def parse_element_tag(el, path):
    matched = el.xpath(path)
    if matched:
        return matched[0].strip()
    return ''

lst = []
paths = './/h3/text()', './@href', './/img/@src'
for el in l.cssselect('div ul.svtGridBlock li div a'):
    lst.append(tuple(parse_element_tag(el, path) for path in paths))

>>> for x in lst: print(x)
('Barn', '/kategorier/barn', '/public/images/categories/kat-barn.png')
(u'Dokument\xe4r', '/kategorier/dokumentar', '/public/images/categories/kat-dokumentar.png')
('Film & Drama', '/kategorier/filmochdrama', '/public/images/categories/kat-filmochdrama.png')
(u'Kultur & N\xf6je', '/kategorier/kulturochnoje', '/public/images/categories/kat-kulturochnoje.png')
('Nyheter', '/kategorier/nyheter', '/public/images/categories/kat-nyheter.png')
(u'Samh\xe4lle & Fakta', '/kategorier/samhalleochfakta', '/public/images/categories/kat-samhalleochfakta.png')
('Sport', '/kategorier/sport', '/public/images/categories/kat-sport.png')
('', '/kategorier/oppetarkiv', '/public/images/categories/kat-oppetarkiv.png')

相关问题 更多 >