Python,DataFrameでcsvにカラムを追加して上書き保存

Question

以下のようなcsvがあった時に、DataFrameとしてcsvを読み込み、
カラムを追加して上書き保存するコードを書きたいのですがどのように書けばスマートでしょうか？

<a.csv(処理前)>
※先頭行はカラム名
A,B,C
p,0,10
q,3,20
r,5,50

<a.csv(処理後)>
A,B,C,D,E
p,0,10,x,1
q,3,20,y,2
r,5,50,z,3

Answer 1

普通にカラムに代入してみました。

import pandas as pd

df = pd.read_csv('a.csv')
df['D'] = ['x', 'y', 'z']
df['E'] = [1, 2, 3]
df.to_csv('a.csv', index=None)

既存のファイルを上書きするのは自己の元なので、出力先はa.csvにしないほうがよいと思います。

Comments

[Obayashi]

言われてみれば簡単なことでしたが助かりました。ありがとうございました。

Answer 2

上書き（追記）というより、いったん全読込→列を追加→全保存という流れでよいと思います。

import pandas as pd

# 読み込み
df = pd.DataFrame([['p',0,10],['q',3,20],['r',5,50]],columns=['A','B','C'])

# 列の追加
df['D'] = ['x','y','z']
df['E'] = [1,2,3]

# 保存
df.to_csv( 'df.csv', index=False )