csvファイルから特定の列の抽出と出力

前提・実現したいこと

はじめて質問します。初心者です。
データ量が多く整理したい為に、csvファイルを使用してデータを読み込みたいと思っております。
pandas等をつかわない方法も試してみようと思ったのですが書き出しができません。
あるcsvファイルの5,6,7列目を取り出し、新しいcsvファイルに入れたいです。

発生している問題・エラーメッセージ

読みこみまではできています。

with open('新しいファイル.csv', 'w') as file:
    writer = csv.writer(file, lineterminator='\n')
    writer.writerow([line[5]])

としたところ、最後の行数だけでしかも1文字ずつセルに入って出てきてしまいました。
上でいう5列目の要素をn行抜き出して新しいファイルを作成したいです。
検索では、一つ一つ要素を入れていく出力ばかりでわかりませんでした。

### やりたいこと
[あ,い,う,え,お]
[か,き,く,け,こ]
[さ,し,す,せ,そ]

というファイルから
with open("あいうえお.csv",'r') as f:
   line = f.readline()
   while(line):
       line = line.split(",")
       print(line[3])
       line = f.readline()
で例えば3,4,5列目を取り出します。
う,え,お
く,け,こ
す,せ,そ


このう、く、すを
[う,え,お]
[く,け,こ]
[す,せ,そ]

としてあたらしいファイルにしたいです。
データが多めなので
writer.writerow(['う', 'え', 'お'])
で全部入れるのはできないので、そのまま入れたいです。

LouiS0616

2018/10/10 06:15

ダミーでもいいのでサンプルのcsvファイルと、それを利用した際に期待する実行結果を追記してください。

行動規範の内容に同意します

回答1件

ベストアンサー

[あ,い,う,え,お]
[か,き,く,け,こ]
[さ,し,す,せ,そ]

というファイル

それはcsvでは無いですね。独自形式です。

このう、く、すを ... としてあたらしいファイルにしたいです。

両端の [ ] 及び改行文字を排除し、スライスを利用します。

Python
1with open('src.txt') as fin,    \
2        open('dst.txt', 'w') as fout:
3        
4    for line in map(str.rstrip, fin):
5        elems = line.strip('[]').split(',')
6        print(f'[{",".join(elems[2:])}]', file=fout)

実行例: Wandbox

コメントを受けて

合法なcsv形式なのですね。

Python
1with open('src.csv', newline='') as fin,    \
2        open('dst.csv', 'w', newline='') as fout:
3        
4    reader = csv.reader(fin)
5    writer = csv.writer(fout)
6    
7    for line in reader:
8        writer.writerow(line[2:])
9
10    # 上記ループを丸ごと writer.writerows(line[2:] for line in reader) に置き換えても可