从嵌套循环写入 output 到 csv ？

Question

I have several loops within a larger loop, and I need the csv file to be written to twice:

UPDATED, more detailed example

with open("file.csv", "w", encoding="utf-8", newline="") as file:
    writer = csv.writer(file)
    while True:
        try:
            element = driver.find_element_by_xpath(xpath)
            if element.is_displayed():
                nextbtn=driver.find_element_by_xpath(xpath).click()
                object=driver.find_element_by_xpath(xpath)
                writer.writerows(object)
        except NoSuchElementException:
            urls=driver.find_elements_by_xpath(xpath)
            urls= [url.get_attribute('href') for url in urls]

            for url in urls:
                driver.get(url)
                object = driver.find_element_by_xpath(xpath)
                writer.writerows(object)

Essentially what the code is doing is testing to check if an element is present. If the element is present, the next button is clicked. Once the next button is clicked, a specific element, object , is scraped and written to csv.

If the certain element is not present, the code then collects a list of urls, and for each url, it scrapes the element object and prints it to csv.

Regardless of which page is navigated to, I want the object element. However, the problem lies in writing to csv, as the results from try are not being written to the csv. Although the location of object is the same for both pages, the value of object is different, depending on the page.

How would I rewrite this code such that both results are written in the same.csv?

Answer 1

For anyone else that runs into the same issue, the problem was NOT the syntax writing to.csv. To write to a.csv in a nested loop, the syntax I posted above is correct.

The issue in this case was that the element object I was trying to find in the try loop was not being found, and was therefore writing empty rows, whereas the object was being found in the except loop. I used an if/else statement within the try to write a cell "element not found" to csv. Alternatively, you could just leave it blank.