python - 如何使用 pandas 读取 csv、附加新数据以及写入新的 csv

Question

我以前没有使用过 Pandas，看起来我需要一些初步的帮助。我在任何地方都找不到这个具体的例子。

我有一个 csv 文件，比如 file1.csv 如下:

ID     value1     value2
1       100        200
2       101        201

我需要一次从 file1.csv 中读取 1 行，附加 2 个新列信息/数据，然后将所有内容写入一个名为 file2.csv 的新文件。 file2.csv 应该如下所示:

ID     value1     value2     value3     value4
1       100        200        10         20
2       101        201        11         21

任何人都可以指导或给出一个简短的例子来说明如何做到这一点(读取 file1，附加新数据(value3 和 value4 列)，并将其写入 file2)？

附录:我需要一次从 file1 读取 1 行并一次将 1 行写入 file2。

Answer 1

以下将加载 file1.csv，添加列 'value3' 和 'value4' 并将生成的数据帧输出为 csv .

import pandas as pd

df = pd.read_csv('file1.csv')
df['value3'] = [10, 11]
df['value4'] = [20, 21]
df.to_csv('file2.csv')

file1.csv 的内容:

ID,value1,value2
1,100,200
2,101,201

file2.csv 的内容:

,ID,value1,value2,value3,value4
0,1,100,200,10,20
1,2,101,201,11,21