既存のcsvファイルの右に、もう一列加えたいです

Question

実現したいこと

test.csvは以下のようなファイルです。

x	y
0	2
1	3

y列の隣にA=(y/2)^2の値を書き込んだ列を追加したいです。つまり、欲しい結果は以下の通りです。

x	y	A
0	2	1
1	3	2.25

「該当のソースコード」の出力result.csvは

x	y	A
[0 1]	[2 3]	[1.   2.25]

です。どなたか「該当のソースコード」を直してください。

該当のソースコード

import pandas as pd

df = pd.read_csv('test.csv')
 
x = df['x'].values
y = df['y'].values

num = 2
A = [0 for i in range(num)]
A = (y/2)**2

filename = 'result.csv'
file = open(filename,'w')
file.write('x'+'\t'+'y'+'\t'+'A'+'\n')
file.write(str(x)+'\t'+str(y)+'\t'+str(A)+'\n')
file.close()

試したこと

A=(y/2)^2の値を書き込んだファイルを別に用意して、そのファイルとtest.csvを横に結合すれば良いとも思いましたが、そもそもA=(y/2)^2の値を書き込んだファイルが

A
[1.   2.25]

となってしまいました。

num = 2
A = [0 for i in range(num)]
A = (y/2)**2

filename = 'A.csv'
file = open(filename,'w')
file.write('A'+'\n')
file.write(str(A)+'\n')
file.close()

Answer 1

出力がTAB区切り(TSV)になっていますので、ファイル名を result.tsv に変更しています。

import pandas as pd

df = pd.read_csv('test.csv')
df['A'] = (df['y'] / 2) ** 2

filename = 'result.tsv'
df.to_csv(filename, sep='\t', index=False)

Comments

[yyicp]

tsvで保存すると、result.tsvを開くときにアプリを選ぶようウィンドウが開いて、そこでメモ帳を選択すると望み通りの画面が表示されました。私はtsvファイルを初めて扱いました。

一方、csvで保存して①excelで開いたときに、区切り位置をタブにすると、望み通りの画面が表示されました。この状態になってから再度保存しました。①の操作をやらずに済むようにcsvで保存することはできないでしょうか。

[melian]

そうですね、その場合はタブではなくカンマ区切りで保存すればよいかと思います。(ファイル名をresult.csvに戻します)

filename = 'result.csv'
df.to_csv(filename, index=False)

[yyicp]

sep=','で解決しました。
どうもありがとうございました。