Reading and writing variables from CSV file in Python (Selenium)

Question

I'm having some difficulties with my code - wondering if anyone could help me as to where I'm going wrong.

The general syntax of the goal I'm trying to achieve is:

1. Get user input
2. Split input into individual variables
3. Write variables (amend) to 'data.csv'
4. Read variables from newly amended 'data.csv'
5. Add variables to list
6. If variable 1 <= length of list, #run some code
7. If variable 2 <= length of list, #run some code

Here is my python code:

from selenium import webdriver
import time
import csv

x = raw_input("Enter numbers separated by a space")
integers = [[int(i)] for i in x.split()]

with open("data.csv", "a") as f:
   writer = csv.writer(f)
   writer.writerows(integers)

with open('data.csv', 'r') as f:
   file_contents = f.read()
   previous_FONs = file_contents.split(' ')

if list.count(integers[i]) == 1:
   #run some code

elif list.count(integers[i]) == 2:
   #run some code

The error message I'm receiving is TypeError: count() takes exactly one argument (0 given)

Answer 1

Because of the following line

integers = [[int(i)] for i in x.split()]

you're creating a list of lists. Therefore you're passing lists to the count method. Try this one:

integers = [int(i) for i in x.split()]

---

Edit: Based on your explanation what you want to achieve, this code should do it:

import csv

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

with open('data.csv', 'a', newline='') 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 each of the user-input numbers
for FON in new_FONs:

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

    if FON_count == 1:
        print(f'{FON[0]} occurs once')
        # do stuff

    elif FON_count == 2:
        print(f'{FON[0]} occurs twice')
        # do stuff

I have changed the name of the list read from CSV to all_FONs just to remind that this contains the old entries and also the new once (as we wrote them into the file before reading).

In addition you need to convert the entries as when reading from CSV you get strings not integers, what would make the comparison difficult. Maybe the whole conversion to int is not necessary, just work on strings. But that depends on what you need.

---

Edit2: Sorry forgot to change from input to raw_input for Python 2.7 :)