將csv文件存儲在數組中，刪除第一列[Python]

Question

我想在刪除第一列的python中將csv文件存儲在數組中。 為了存儲它，我只編寫了這兩行代碼

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

而且有效。 從這一點上，我如何擺脫第一列？

Answer 1

只需使用numpy.delete函數。 如下所示將其添加到代碼中：

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的delete文檔在此處 。