Python从CSV文件提取指定数据并保存为新CSV文件

从CSV文件提取指定数据并保存为新文件

本文将教你如何使用Python从名为'jichu.csv'的CSV文件中提取'Alpha'、'Beta'、'CDtot'和'CL'列的数据，并将结果保存到一个名为'jieguo.csv'的新CSV文件中。

以下是完整的Python代码：

import csv

# 打开CSV文件
with open('jichu.csv', 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

# 寻找指定列的索引
header_row = data[0]
alpha_index = header_row.index('Alpha')
beta_index = header_row.index('Beta')
cdtot_index = header_row.index('CDtot')
cl_index = header_row.index('CL')

# 初始化结果字典
results = {'Alpha': None, 'Beta': None, 'CDtot': None, 'CL': None}

# 遍历文件内容，并提取结果
for row in data:
    if 'Results_Name' in row and 'VSPAERO_History' in row:
        # 找到目标行
        results['Alpha'] = row[alpha_index]
        results['Beta'] = row[beta_index]
        results['CDtot'] = row[cdtot_index]
        results['CL'] = row[cl_index]

# 将结果保存到新的CSV文件
output_data = [['Alpha', 'Beta', 'CDtot', 'CL'], [results['Alpha'], results['Beta'], results['CDtot'], results['CL']]]
with open('jieguo.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(output_data)

print('结果已保存到 jieguo.csv 文件中。')

代码解释：

导入csv模块: 用于处理CSV文件。
打开CSV文件: 使用with open()函数打开'jichu.csv'文件，并使用csv.reader()读取文件内容。
寻找指定列的索引: 获取第一行作为表头，并使用index()方法找到目标列的索引。
初始化结果字典: 创建一个字典用于存储提取的结果。
遍历文件内容并提取结果: 遍历每一行数据，找到包含'Results_Name'和'VSPAERO_History'的行，提取对应列的数据并存储到结果字典中。
将结果保存到新CSV文件: 使用csv.writer()将结果写入名为'jieguo.csv'的新文件中。

请注意:

将代码中的'jichu.csv'替换为你实际的CSV文件名。
根据你的需求修改要提取的列名和结果保存方式。

希望这段代码能够帮助你从CSV文件中提取指定数据并保存到新的CSV文件中！