Converting CSV column to list of lists in python

Question

I have a csv column with some labels in it. I need to convert it to list of lists.
ex:

type1(heading)  
string1  
string2  
string3  
string4

converting this directly to,

type1 = [[string1],[string2],[string3],[string4]]

Used pandas lib to read csv and then converted the column to a list. Then, I am having problem converting it to list of lists. @AnandSKumar — javi_p
– javi_p, Commented Oct 12, 2015 at 6:49
If you give an actual example piece of the file it is more worth than just some text. We need to know what differentiates the text from the headings — holroy
– holroy, Commented Oct 12, 2015 at 14:56

Giovanni Oliveira · Accepted Answer · 2015-10-12 06:36:24Z

1

It seems python has a native lib to deal with such tasks. Maybe you should take a look at https://docs.python.org/2/library/csv.html

answered Oct 12, 2015 at 6:36

Giovanni Oliveira

772 silver badges10 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

kkpattern · Accepted Answer · 2015-10-12 07:22:18Z

1

You can try the csv module. For example, given a test.txt file:

header_a,header_b,header_c
1,2,3
4,5,6
7,8,9

You can write:

import csv

with open("test.txt", "r") as f:
    reader = csv.reader(f)
    headers = list(reader.next())
    result = [[list(c) for c in row] for row in reader]
    for i, header_name in enumerate(headers):
        print header_name, [row[i] for row in result]

The result will be:

header_a [['1'], ['4'], ['7']]
header_b [['2'], ['5'], ['8']]
header_c [['3'], ['6'], ['9']]

The document of csv module is here

edited Oct 12, 2015 at 7:22

answered Oct 12, 2015 at 6:34

kkpattern

9982 gold badges10 silver badges32 bronze badges

2 Comments

javi_p Over a year ago

Thank you! But I need to convert elements under same header to a list of list.

kkpattern Over a year ago

@javi_parni: I've updated my answer, see if that's what you need.

Collectives™ on Stack Overflow

Converting CSV column to list of lists in python

2 Answers 2

Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related