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_file、output_xlsx_file和columns_to_extract变量的值,然后运行代码即可完成从CSV文件中读取特定列并保存到XLSX文件的操作。
野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892

- 上一篇:python读取大型(几个G)的csv
- 下一篇:用excel处理csv文件
