Python code
data.csv