Выпустить список извлечения из файла CSV в Python (Selenium) - PullRequest
0 голосов
/ 10 января 2020

Мой код принимает входные данные и успешно записывает их в файл «data.csv», но всякий раз, когда в качестве начального ввода вводится более одного числа (например, «98765 12345»), я получаю IndexError: list index out of range в самой последней строке.

Кто-нибудь может увидеть, где я иду не так?

x = raw_input('Enter numbers separated by a space: ')
new_FONs = [[int(i)] for i in x.split()]

with open('data.csv', 'a+') as f:
    writer = csv.writer(f)
    writer.writerows(new_FONs)

with open('data.csv', 'r') as f:
    all_FONs_str = [line.split() for line in f]
    all_FONs = [[int(FON[0])] for FON in all_FONs_str]

for FON in new_FONs:

    # Count the occurence of this number in the CSV file
    FON_count = all_FONs.count(FON)

    if FON_count == 1:
        print('once')

    elif FON_count == 2:
        print('twice')

1 Ответ

0 голосов
/ 10 января 2020

Ограничение последнего l oop предпоследним элементом должно сделать:

x = raw_input('Enter numbers separated by a space: ')
new_FONs = [[int(i)] for i in x.split()]

with open('data.csv', 'a+') as f:
    writer = csv.writer(f)
    writer.writerows(new_FONs)

with open('data.csv', 'r') as f:
    all_FONs_str = [line.split() for line in f]
    all_FONs = [[int(FON[0])] for FON in all_FONs_str[:-1]]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...