当前位置:首页python > 正文

python在csv中添加一列数据

作者:野牛程序员:2023-07-24 07:56:08python阅读 3679

在 Python 中,可以使用标准库中的 csv 模块来读取和写入 CSV 文件,并使用 pandas 库来处理数据和添加列。下面将提供两种方法,一种使用 csv 模块,另一种使用 pandas 库。

方法一:使用 csv 模块

import csv

# 读取原始CSV文件的数据
input_file = 'input.csv'
output_file = 'output.csv'

with open(input_file, 'r', newline='') as infile, open(output_file, 'w', newline='') as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)

    # 添加列名到头部
    header = next(reader)
    header.append('New Column')
    writer.writerow(header)

    # 添加数据到新列
    for row in reader:
        new_data = 'New Value'  # 在这里替换成想要添加的新数据
        row.append(new_data)
        writer.writerow(row)

方法二:使用 pandas

如果数据集较大,pandas 是一个更方便和高效的选择。

import pandas as pd
# 读取CSV文件
input_file = 'input.csv'
output_file = 'output.csv'
df = pd.read_csv(input_file)
# 添加新列
new_data = ['New Value'] * len(df)  # 在这里替换成想要添加的新数据
df['New Column'] = new_data
# 将数据写入新的CSV文件
df.to_csv(output_file, index=False)

请注意,在以上两种方法中,需要将 'input.csv' 替换成要读取的CSV文件名,而将 'output.csv' 替换成要保存的新CSV文件名。在方法一中,替换 'New Value' 为你想要添加到新列中的实际数据。在方法二中,替换 ['New Value'] * len(df) 为想要添加到新列中的实际数据列表。


野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击