将列附加到 python 中的现有 csv 文件

Question

以前有人问过这个问题，但这些解决方案对我不起作用。我有一个 csv 文件，example.csv，数据格式如下：

grid,u,lambda
0.0,0.0,-93000
0.01,-0.01,-93000
0.02,0.05,-93000

现在我有一个函数 f.fraction('function') returns 我 [1.00000000e+00 1.00000000e+00 1.00000000e+00]。我想将其附加到同一个文件 example.csv。最终结果：

grid,u,lambda,new_function
0.0,0.0,-93000,1.00000000e+00
0.01,-0.01,-93000,1.00000000e+00
0.02,0.05,-93000,1.00000000e+00

我该怎么做？

Answer 1

您可以先使用 pandas 读取文件，追加新列然后保存。

import pandas as pd
df = pd.read_csv('example.csv')
df['new_function'] = f.fraction('function')
df.to_csv('example.csv')

Append a column to an existing csv file in python