2016-12-15 16 views
0

最初の列を削除するPythonで配列にCSVファイルを保存します。それを保存するために、私はちょうどこれらの2行のコードを書いています。最初の列を削除して配列にcsvファイルを格納する[Python]

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

それは働きました。この時点から、どのようにして最初の列を取り除くことができますか?

+1

作業しているカラムの数( 'num_cols')が分かった場合は、' genfromtext'メソッドで直接カラムを取得できます: 'data = genfromtxt( 'SP500Index.csv'、デリミタ= '、'、usecols =範囲(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です。

関連する問題