Data Collection Python Script for Raspberry Pi with Sense Hat

datacollection.py

#!/usr/bin/python3

from ev3dev.ev3 import *
import csv
import time

filename = input('Enter a name for the file: ')
filename = filename+".csv"

cl = ColorSensor()
cl.mode='RGB-RAW'

with open(filename, 'w') as logfile:
        fieldnames = ['red', 'green', 'blue']
        logwriter = csv.DictWriter(logfile, fieldnames=fieldnames)

        logwriter.writeheader()

        for x in range(0,1500):
                red = cl.value(0)
                green = cl.value(1)
                blue = cl.value(2)

                logwriter.writerow({
                'red' : red,
                'green' : green,
                'blue' : blue
                })

                time.sleep(.1)