当前位置:首页python > 正文

Python从CSV文件中读取特定列并转存至xlsx文件

作者:野牛程序员:2023-07-24 07:51:59python阅读 3066

要从CSV文件中读取特定列并转存至XLSX文件,你可以使用Python的pandas库来完成这个任务。pandas是一个强大的数据处理库,它可以轻松地读取CSV文件、操作数据并将数据保存到Excel文件中。

首先,请确保已经安装了pandas库。如果没有安装,可以使用以下命令来安装:

pip install pandas

接下来,可以使用以下Python代码来实现从CSV文件中读取特定列,并将这些列保存到XLSX文件中:

import pandas as pd

def extract_columns_from_csv(input_file, output_file, columns_to_extract):
    try:
        # 使用pandas读取CSV文件
        df = pd.read_csv(input_file)
        
        # 从DataFrame中选择指定的列
        selected_columns_df = df[columns_to_extract]
        
        # 将选择的列保存到XLSX文件
        selected_columns_df.to_excel(output_file, index=False)
        
        print(f"成功从 {input_file} 中提取并保存了以下列到 {output_file}: {', '.join(columns_to_extract)}")
    
    except Exception as e:
        print("出现错误:", e)

# 输入CSV文件路径
input_csv_file = "input.csv"

# 输出XLSX文件路径
output_xlsx_file = "output.xlsx"

# 需要提取的列名
columns_to_extract = ["Column1", "Column2", "Column5"]

# 调用函数进行提取和保存
extract_columns_from_csv(input_csv_file, output_xlsx_file, columns_to_extract)

在上述代码中,extract_columns_from_csv函数接受三个参数:输入CSV文件路径input_file,输出XLSX文件路径output_file和一个列表columns_to_extract,其中包含需要从CSV文件中提取的列名。

请根据实际情况修改input_csv_fileoutput_xlsx_filecolumns_to_extract变量的值,然后运行代码即可完成从CSV文件中读取特定列并保存到XLSX文件的操作。


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

最新推荐

热门点击