Store csv file in array removing first column [Python]

Question

I want to store a csv file in an array in python removing the first column. In order to store it I just write these 2 lines of code

from numpy import genfromtxt
def csvToArray():
    data = genfromtxt('SP500Index.csv', delimiter=',')
    return array

and it worked. From this point how can I get rid of the first column?

If you happen to know the number of columns you're working with (num_cols), then you can directly get the columns you want through the genfromtext method: data = genfromtxt('SP500Index.csv', delimiter=',', usecols = range(1,num_cols)). — Abdou
– Abdou, Commented Dec 15, 2016 at 12:42

Community · Accepted Answer · 2017-08-24 17:11:39Z

1

Just use the numpy.delete function. Add it to the code as follows:

import numpy as np

def csvToArray():
    data = np.genfromtxt('SP500Index.csv', delimiter=',')
    data = np.delete(data, 0, 1) # the first input is the array, second is
                          #   which row/column to delete, third
                          #   determines if it is row (0) or column (1)
    return data #not sure why it had array before

Numpy's delete documentation is here.

edited Aug 24, 2017 at 17:11

CommunityBot

11 silver badge

answered Dec 15, 2016 at 12:35

AER

1,5411 gold badge19 silver badges39 bronze badges

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

Collectives™ on Stack Overflow

Store csv file in array removing first column [Python]

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related