web-dev-qa-db-ja.com

CSVファイルの列を削除する方法は？

このサイトの複数のユーザーからの入力を使用してpythonを使用してcsvを作成することができました。あなたの投稿に感謝の意を表したいと思います。

私のinput.csvは次のようになります：

day,month,year,lat,long
01,04,2001,45.00,120.00
02,04,2003,44.00,118.00

「年」列とそのすべてのエントリを削除しようとしています。合計で1960〜2010年の範囲で40以上のエントリがあります。

pythoncsvrow-removal
25
Jeff
import csv
with open("source","rb") as source:
    rdr= csv.reader( source )
    with open("result","wb") as result:
        wtr= csv.writer( result )
        for r in rdr:
            wtr.writerow( (r[0], r[1], r[3], r[4]) )

ところで、forループは削除できますが、実際には単純化されていません。

        in_iter= ( (r[0], r[1], r[3], r[4]) for r in rdr )
        wtr.writerows( in_iter )

また、列を削除するための要件にハイパーリテラルの方法で固執することができます。複数の列の削除には適用されないため、これは一般的に悪いポリシーであると思います。 2番目を削除しようとすると、位置がすべてシフトし、結果の行が明らかではないことがわかります。しかし、1つの列に対してのみ、これは機能します。

            del r[2]
            wtr.writerow( r )
42
S.Lott

Pandasモジュールの使用ははるかに簡単になります。

import pandas as pd
f=pd.read_csv("test.csv")
keep_col = ['day','month','lat','long']
new_f = f[keep_col]
new_f.to_csv("newFile.csv", index=False)

そして、ここに簡単な説明があります：

>>>f=pd.read_csv("test.csv")
>>> f
   day  month  year  lat  long
0    1      4  2001   45   120
1    2      4  2003   44   118
>>> keep_col = ['day','month','lat','long'] 
>>> f[keep_col]
    day  month  lat  long
0    1      4   45   120
1    2      4   44   118
>>>
25
SunilThorat

Dictを使用して見出しを取得し、ループスルーすることで、必要なものをきれいに取得できます。

import csv
ct = 0
cols_i_want = {'cost' : -1, 'date' : -1}
with open("file1.csv","rb") as source:
    rdr = csv.reader( source )
    with open("result","wb") as result:
        wtr = csv.writer( result )
        for row in rdr:
            if ct == 0:
              cc = 0
              for col in row:
                for ciw in cols_i_want: 
                  if col == ciw:
                    cols_i_want[ciw] = cc
                cc += 1
            wtr.writerow( (row[cols_i_want['cost']], row[cols_i_want['date']]) )
            ct += 1
7
Aimon Bustardo

あなただけの列を直接削除することができます

del variable_name['year']
1
ankur

csvパッケージを使用してcsvファイルを反復処理し、目的の列を別のcsvファイルに出力できます。

以下の例はテストされていないため、ソリューションを説明する必要があります。

import csv

file_name = 'C:\Temp\my_file.csv'
output_file = 'C:\Temp\new_file.csv'
csv_file = open(file_name, 'r')
## note that the index of the year column is excluded
column_indices = [0,1,3,4]
with open(output_file, 'w') as fh:
    reader = csv.reader(csv_file, delimiter=',')
    for row in reader:
       tmp_row = []
       for col_inx in column_indices:
           tmp_row.append(row[col_inx])
       fh.write(','.join(tmp_row))
1
aweis

私の頭の外では、これはどんな種類のエラーチェックも何も構成する能力なしでそれをします。それは「読者に任せて」です。

outFile = open( 'newFile', 'w' )
for line in open( 'oldFile' ):
   items = line.split( ',' )
   outFile.write( ','.join( items[:2] + items[ 3: ] ) )
outFile.close()
1
broc

解析されたCSVの保存方法によって異なりますが、通常はdel演算子が必要です。

辞書の配列がある場合：

input = [ {'day':01, 'month':04, 'year':2001, ...}, ... ]
for E in input: del E['year']

配列の配列がある場合：

input = [ [01, 04, 2001, ...],
          [...],
          ...
        ]
for E in input: del E[2]
0
KQ.

試してください：

result= data.drop('year', 1)
result.head(5)
0
AlfhaOmega