pythonでcsv読み込み・新たに出力

Question

name,a,b,c,d,e,f,g
aaa,1,0,0,0,1,0,0
bbb,2,1,0,0,2,0,1
ccc,1,2,0,0,1,0,1

上記のようなcsvがあるときに、python言語を用いて
aaaのbの値を0→1に変更など、特定の数字を指定して値を書き換え、新しいcsvのファイルとして出力したいとき、どのようにすれば良いでしょうか？

Answer 1

panda を使わずにかいてみました。
p.py

import csv

#csvファイル
in_file = "data.csv"
out_file = "data2.csv"

with open(in_file) as f:
    rows = list(csv.DictReader(f))
data = [ {k: v if k == 'name' else int(v) for k, v in row.items()} for row in rows]
# print(data)

data2 = {row["name"]: row for row in data}
# print(data2)

data2["aaa"]["a"] = 99

header = data[0].keys()
with open(out_file, 'w', newline='') as f:
    writer = csv.DictWriter(f, fieldnames=header)
    writer.writeheader()
    for _k, v in data2.items():
      writer.writerow(v)

実行例

参考情報

• Python CSVファイル 読み込みと書き込み

https://qiita.com/yusa0827/items/8f33cb8dbacc9040bc24

• Pythonでcsvファイルを読み込む：ヘッダーのついたcsvはDictReaderが便利

https://www.mathpython.com/ja/python-csv-read/

Answer 2

この手の処理なら、pandasを使うと楽ですね。

import pandas as pd

df = pd.read_csv("sample.csv", index_col=0)

print(df)

print(f"old value: {df.at['aaa', 'b']}")
df.at['aaa', 'b'] = 1
print(f"new value: {df.at['aaa', 'b']}")

print(df)

df.to_csv("sample_new.csv")

Comments

[edamame11]

特定の行・列の値の取得方法が分からなかったので助かりました。
うまいことできました。
ありがとうございます。

Answer 3

まずは、CSVファイルを読み込み、
CSVファイルを書き込むコードを組んでみてはどうでしょう
何事も、かんたんなことから順番にやっていかないとできませんよ

Comments

[edamame11]

ありがとうございます。
そうですね、ちょっと自分で考えてみます。