python csvファイル改行削除

Question

csvファイル
['abc\ndef\nghi','jkl\nmno\npqr','stu\nvwx\nyz']
改行コードを削除したい。
readerで読み込んだファイルは削除できますが、
writerで読み込むとエラーが出ます。

for row in reader:
TypeError: '_csv.writer' object is not iterable

何がおかしいのかご教示ください。

#readerで読み込むと改行を削除できるコード
import csv
list=[]
text=''
with open('example.csv','r',newline='') as f:
reader=csv.reader(f,delimiter=',')
for row in reader:
list+=row
for i in range(len(list)):
text=list[i]
list[i]=text.replace('\n','')
print(list)

結果：['abcdefghi','jklmnopqr','stuvwxyz']

#writerで読み込むとエラーが出るコード
import csv
list=[]
text=''
with open('example.csv','w',newline='') as f:
writer=csv.writer(f,delimiter=',')
for row in writer:
list+=row
for i in range(len(list)):
text=list[i]
list[i]=text.replace('\n','')

結果： for row in reader:
TypeError: '_csv.writer' object is not iterable

Comments

[退会済みユーザー]

writerはファイル書き込み用で読み込み用ではありませんが…

Answer 1

Excelから出力したCSVファイルから、セル内改行を除去したいということであれば以下でできます。

import csv
# 読込
rows = []
with open('in.csv','r') as f:
    reader = csv.reader(f)
    for row in reader:
        # セル内改行を除去
        for i,v in enumerate(row):
            row[i] = v.replace('\n', '')
        rows.append(row)
# 書込
with open('out.csv','w', newline='') as f: # 読込同名ファイルに上書きでもよい
    writer = csv.writer(f, dialect='excel') # dialect='excel' は指定しなくてもよい
    for row in rows:
        writer.writerow(row)

in.csv

ab,"cd(\n)
ef",gh(\r\n)
ij,kl,mn(\r\n)

out.csv

ab,cdef,gh(\r\n)
ij,kl,mn(\r\n)

(\n)は改行、(\r\n)は復帰改行です。

Comments

[watarijin]

ご教示いただきありがとうございます。
解決できました。助かりました。
readerとwriterで同じ動作をさせようとしたところが悪かったです。
私の手続きが多いコードを簡素化してもらいありがとうございます。

Answer 2

example.csvの形式が分からないので

改行コードを削除したい

という点に対しては、読み込んだセルの文字列に対して.slpit("\n")で分割すればいいように思います。

s = ['abc\ndef\nghi','jkl\nmno\npqr','stu\nvwx\nyz']
t = [_s.split("\n") for _s in s]
# [['abc', 'def', 'ghi'], ['jkl', 'mno', 'pqr'], ['stu', 'vwx', 'yz']]

Comments

[watarijin]

リスト内包表記の書き方勉強になります。
ご教示ありがとうございます。

Answer 3

こんなことがしたいのかな？と思ってコードを書いてみました。

example.csv (スクリプト実行前)

abc\ndef\nghi,jkl\nmno\npqr,stu\nvwx\nyz

main.py

from pathlib import Path

filename = 'example'
src_file = Path(f'{filename}.csv')
dst_file = Path(f'{filename}_dst.csv')

with src_file.open() as fin, dst_file.open('w') as fout:
    for row in fin:
        fout.write(row.replace(r'\n', ''))

dst_file.replace(str(src_file))

example.csv (スクリプト実行後)

abcdefghi,jklmnopqr,stuvwxyz

csvで読み取る必要性が感じられなかったので、ただのテキストファイルとして処理しています。

気になったところ

listは**変数名として使ってはいけません。**分かりづらいエラーを引き起こします。

>>> list = list(range(3))
>>> list
[0, 1, 2]
>>>
>>> list2 = list(range(4))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: 'list' object is not callable

他にも max int input id set sum str なども誤って使ってしまいがちです。

質問のコードの書き方について

teratailには、上記のようにコードを見やすく表示する機能があります。
質問編集画面を開き、コードを選択した状態で<code>ボタンを押してください。
特にPythonの場合、インデントが崩れるとコードの意味が変わってしまいます。

Comments

[watarijin]

早速ご指導いただきありがとうございます。
自分でもコードを入れてやってみましたが、この状況ではうまくいくようです。しかし、今回壁にぶつかったのが、エクセルのセル内改行を削除しなければならなかったのが、このやり方ではできないようでした。
説明不足で申し訳ございません。
もう少し試してみます。
どうもありがとうございました。