将csv文件存储在数组中，删除第一列[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?

Answer 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 .