如何在Python中将CSV转换为Excel

CSV和Excel都是常用的数据格式，CSV比起Excel来说更轻量级，但Excel有更强大的功能。在处理数据时，有时候需要将CSV转换为Excel格式，这篇文章将介绍如何在Python中实现这一功能。

1. 安装第三方库

要在Python中将CSV转换为Excel，需要用到第三方库pandas和openpyxl，可以使用pip命令安装：

$ pip install pandas openpyxl

2. 读取CSV文件

使用pandas库中的read_csv函数读取CSV文件，并将数据存放在DataFrame中。

import pandas as pd

# 读取CSV文件
data = pd.read_csv('data.csv')

3. 将数据写入Excel

使用openpyxl库中的Workbook和Worksheet类将数据写入Excel文件。

from openpyxl import Workbook

# 创建一个新的Excel文件
wb = Workbook()

# 创建一个新的工作表
ws = wb.active

# 将数据写入工作表
for r in dataframe_to_rows(data, index=False, header=True):
    ws.append(r)

# 保存Excel文件
wb.save('data.xlsx')

dataframe_to_rows函数是将DataFrame转换为行的生成器函数。在append函数中循环逐行将数据写入Excel文件。

完整代码

import pandas as pd
from openpyxl import Workbook
from openpyxl.utils.dataframe import dataframe_to_rows

# 读取CSV文件
data = pd.read_csv('data.csv')

# 创建一个新的Excel文件
wb = Workbook()

# 创建一个新的工作表
ws = wb.active

# 将数据写入工作表
for r in dataframe_to_rows(data, index=False, header=True):
    ws.append(r)

# 保存Excel文件
wb.save('data.xlsx')

总结

本文介绍了如何在Python中将CSV文件转换为Excel文件。通过使用pandas和openpyxl库，可以实现方便快捷地数据转换。