python：csvファイルを条件によって書き換えたいです

Question

前提・実現したいこと

csvファイルで、下記を追記した上で上書き保存したいと思っております。

・会社　→ 「御中」を追記
・上記以外　→ 「様」を追記

（１）追記まではできたと思うのですが、この内容を保存する方法がわかりませんでした。
教えていただければ幸いです。よろしくお願い致します。

（２）追記する条件を、「会社」があるか無いかではなく、
行（と呼んでよいのかわからないのですが）とすることは可能なのでしょうか。

たとえば、今回のcsvデータで「株式会社山田」と「田中花子」のみ
「さま」を追記したい場合、if文でどのように指定すれば良いのかわからず
こちらも教えていただければ有難いです。どうぞよろしくお願い致します。

該当のソースコード

お名前,電話番号,住所
山田太郎,090-1233-4567,東京都
株式会社山田,03-1111-2222,神奈川県
田中花子,080-1233-4567,千葉県
はな有限会社,045-111-2345,神奈川県
Nancy,050-1234-5678,茨城県

import csv

with open('name.csv', "r") as f:
    s = csv.reader(f)
    h = next(csv.reader(f)) #ヘッダー（住所）をスキップ
    for row in s:
        name = row[0]
        if  '会社' in name:
            name += '　御中'
        else:
            name += '　様' 

Comments

[magichan]

(2)の内容がよくわかりません
予め
['株式会社山田', '田中花子']
のようなデータを準備しておき、それに含まれている場合はと含まれない場合で処理を分けたいということですか？

[kero16]

いえ、このname.csvで、
株式会社山田さま
田中花子さま
それ以外はそのまま変更せずとしたい場合の指定方法を教えていただきたいと思いました。

実際に変更したいデータが違うため、わかりづらい例えになってしまって申し訳ありません。
よろしくお願い致します。

[magichan]

やはりよくわかりません
「株式会社山田」と「田中花子」の２行にのみ「さま」をつけたいのはわかったのですが、その２行はどのように（どのような条件で）指定したいのでしょうか。

[kero16]

「株式会社山田」と「田中花子」は例えでして、
条件を「行」指定にすることは可能かどうかをお聞きしたかったのです。
（このcsvでいえば、3行目と４行目の「お名前」という風に）

「行」という言い方が合っているのかわならなかったため、例えを出してしまったのですが
よけいわかりづらいことになってしまってすみません。

[magichan]

やっと理解しました。行番号で指定したいということですね。

[kero16]

行番号という言い方なのですね！
そうなのです。よろしくお願い致します。

Answer 1

(1) データの保存方法
変更したデータを２次元リストに格納していき、csv.writer() を使って書き込みを行うとよいです

import csv

# 変更したデータを格納するリスト
data = []
with open('name.csv', "r") as f:
    reader = csv.reader(f)
    header = next(reader)
    # ヘッダ行をdataに格納する
    data.append(header)
    for row in reader:
        name = row[0]
        if  '会社' in name:
            row[0] += '　御中'
        else:
            row[0] += '　様'
        # 変更した行データをdataに格納する
        data.append(row)

# データ書き込み部
with open('new_name.csv', "w") as f:
    writer = csv.writer(f)
    writer.writerows(data)

(2) 行番号を指定してデータを変更する方法
CSVデータの行毎にループ処理をする際に enumerate() を使って現在何行目を処理を行っているかをカウントして、その値を条件に処理を行うと良いかと思います。

import csv

data = []
with open('name.csv', "r") as f:
    reader = csv.reader(f)
    header = next(reader)
    data.append(header)
    # noに行番号が入ります（行番号は1からスタートとします）
    for no, row in enumerate(reader, start=1):
        name = row[0]
        # no が 2と3のときのみに処理を行います。
        if no in [2,3]:
            row[0] += '　さま'
        data.append(row)

with open('new_name.csv', "w") as f:
    writer = csv.writer(f)
    writer.writerows(data)

---

一応補足として。。
この手の処理でしたら、pandas を使うことで簡単に書けますのでオススメしておきます。

(1)

import pandas as pd

df = pd.read_csv('name.csv')
df.loc[df['お名前'].str.contains('会社'), 'お名前'] += '　御中'
df.loc[~df['お名前'].str.contains('会社'), 'お名前'] += '　様'
df.to_csv('new_name.csv', index=False)

(2)

import pandas as pd
df = pd.read_csv('name.csv')
df.loc[[1,2], 'お名前'] += '　さま'
df.to_csv('new_name.csv', index=False)

(pandasの場合index値は0から開始なので[1,2]となります)

Comments

[kero16]

ありがとうございます！
（１）（２）ともに求めていた書き換えができました！！

またpandasを使う方法も教えてくださいましてありがとうございます。
やり方を検索していた時に、よく出てきた名前でしたが理解できなかったところ
実際にやりたかった内容をコードにしていただいたので、改めて勉強したいと思います。
pandasのインストールも無事できました。

本当にありがとうございました。