2016-12-15 81 views
0

我想将一个csv文件存储在python数组中删除第一列。为了存储它,我只写了这两行代码在阵列中存储csv文件删除第一列[Python]

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

它工作。从这一点我怎么能摆脱第一列?

+1

如果您碰巧知道您正在使用的列数('num_cols'),那么您可以通过'genfromtext'方法直接获取所需列:'data = genfromtxt('SP500Index.csv', delimiter =',',usecols = range(1,num_cols))'。 – Abdou

+0

谢谢!我不知道usecols参数。 – Pino

回答

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文档here