使用 Pandas 的 DataFrame 对象的 'assign()' 方法可以添加一列数据,然后将修改后的 DataFrame 写入原始 csv 文件。

假设原始 csv 文件名为 'data.csv',要在其中追加一列名为 'new_column' 的数据,可以按照以下步骤进行操作:

  1. 使用 Pandas 的 'read_csv()' 函数读取原始 csv 文件,生成一个 DataFrame 对象。
import pandas as pd

df = pd.read_csv('data.csv')
  1. 使用 'assign()' 方法为 DataFrame 对象添加新列数据。
df = df.assign(new_column=[1,2,3,4,5])

上述代码中,'new_column' 列的数据为一个列表,长度与原始 DataFrame 的行数相同。

  1. 将修改后的 DataFrame 写入原始 csv 文件,使用 'to_csv()' 方法。
df.to_csv('data.csv', index=False)

上述代码中,'index=False' 参数表示不将 DataFrame 的行索引写入 csv 文件。

完整代码如下:

import pandas as pd

df = pd.read_csv('data.csv')
df = df.assign(new_column=[1,2,3,4,5])
df.to_csv('data.csv', index=False)
Pandas 追加 CSV 文件列:使用 assign() 方法

原文地址: https://www.cveoy.top/t/topic/oSPb 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录