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

- 上一篇:python写入csv指定列
- 下一篇:python创建csv文件并写入
