在陣列中存儲csv文件刪除第一列[Python]

我想將一個csv文件存儲在python數組中刪除第一列。爲了存儲它，我只寫了這兩行代碼在陣列中存儲csv文件刪除第一列[Python]

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

它工作。從這一點我怎麼能擺脫第一列？

來源

2016-12-15 Pino

如果您碰巧知道您正在使用的列數（'num_cols'），那麼您可以通過'genfromtext'方法直接獲取所需列：'data = genfromtxt（'SP500Index.csv'， delimiter ='，'，usecols = range（1，num_cols））'。 – Abdou

謝謝！我不知道usecols參數。 – Pino

只需使用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文檔here。

來源

2016-12-15 12:35:15 AER

在陣列中存儲csv文件刪除第一列[Python]

回答

相關問題