Have a question. If I run this code, I get exactly what I have in the CSV. My target is to get a text/data/csv file which would look like this: ['red', 'green', 'blue'] meaning:
Is it possible to do it through Python? Is there any online materials I can look into?
The csv file is read sequentially, that is, you will always get the data back one row at a time. However, you can build an array of just the values you want as you read the file and discard the rest of the data.
import csv
with open('example.csv') as csvfile:
colours = []
for row in csv.reader(csvfile, delimiter=','):
if len(row) <= 3:
continue
colours.append(row[3])
You can achieve this code using pandas. Try this code:
import pandas as pd
df = pd.read_csv("input csv file")
df = df.T
df.to_csv("output csv file")
Update if this is what you are looking for
Without using any extra libraries, you could write a function that takes a specific column, and returns that column as an array.
Such a function might look like this.
def column_to_row(col_num, csv):
return list(row[col_num] for row in csv)
Then you can extract whatever column you want, or iterate through the whole csv like this.
new_csv = []
for i in range(0, len(csv[0]):
new_csv = new_csv.append(column_to_row(i, csv))
csv.reader which is an iterator.