如何在循环中使用范围:Python

Question

这将打印出10个以上的链接，如何将其限制为仅获取前3个链接？

def getlinks(self):
self.driver.get('https://www.youtube.com/results?search_query=iphone&sp=EgQIBRAB')
the_links = self.driver.find_elements(By.ID, "video-title")
sleep(5)
for link in the_links:
self.links.append(link.get_attribute('href'))
for link in self.links:
print(link)

Answer 1

尝试for link in self.links[:3]。

也可以使用等效的表达式for link in self.links[0:3]。