How can I turn a csv file into a list of list in python

Question

I want to be able to turn csv file into a list of lists with the column values for each list. For example:

6,2,4
5,2,3
7,3,6

into

[[6,5,7],[2,2,3],[4,3,6]]

Ive only managed to open the file and only having success printing it as rows

with open(input,'rb') as csvfile:
        csv_file = csv.reader(csvfile)

        header = csv_file.next() 

        raw_data = csv_file

Answer 1

In case you sure it's fixed number of items in each row, you can use zip:

import csv

with open('test.csv') as csvfile:
    rows = csv.reader(csvfile)
    res = list(zip(*rows))
    print(res)
    # [('6', '5', '7'), ('2', '2', '3'), ('4', '3', '6')]

Or in case it's different number of items in row:

6,2,4
5,2
7

Use zip_longest and filter:

import csv
from itertools import zip_longest

with open('test.txt') as csvfile:
    rows = csv.reader(csvfile)

    res = list(zip_longest(*rows))
    print(res)
    # [('6', '5', '7'), ('2', '2', None), ('4', None, None)]

    res2 = [list(filter(None.__ne__, l)) for l in res]
    print(res2)
    # [['6', '5', '7'], ['2', '2'], ['4']]