博客

python脚本实现数据导出excel格式的简单方法(推荐)

  • Post category:Python

下面我将为您详细讲解一下“python脚本实现数据导出excel格式的简单方法(推荐)”的完整实例教程。

准备工作

在开始编写脚本之前，我们需要安装一个Python库，它可以帮助我们导出excel格式的文件，这个库叫做pandas。所以首先需要使用pip安装pandas库。

pip install pandas

安装完成后，我们就可以开始编写Python脚本了。

示例1

以下是一个简单的例子，展示了如何使用pandas将数据导出到excel文件。假设我们要将下面的两个列表导出到excel文件中：

names = ["John", "Monica", "David"]
ages = [25, 32, 47]

我们可以使用以下代码：

import pandas as pd

data = {"Name": names, "Age": ages}
df = pd.DataFrame(data)

writer = pd.ExcelWriter('output.xlsx')
df.to_excel(writer, index=False)
writer.save()

在以上代码中，我们首先创建了一个字典data来存储我们要输出的数据，然后使用pandas的DataFrame函数将数据转换为dataframe对象，接着，我们创建了一个名为writer的ExcelWriter对象，然后通过调用to_excel方法来将数据导出为excel文件，并使用index=False来防止pandas将自动添加索引列。最后，我们调用save方法来保存文件。

示例2

以下是另一个示例，展示了如何从数据库中读取数据，并将其导出到excel文件中。假设我们连接到名为’mydatabase‘的mysql数据库，并从’students‘表中读取所有记录的数据。

import pandas as pd
import mysql.connector

mydb = mysql.connector.connect(
  host="localhost",
  user="username",
  passwd="password",
  database="mydatabase"
)

mycursor = mydb.cursor()

mycursor.execute("SELECT * FROM students")

data = mycursor.fetchall()

df = pd.DataFrame(data, columns=["ID", "Name", "Age"])

writer = pd.ExcelWriter('output.xlsx')
df.to_excel(writer, index=False)
writer.save()

mycursor.close()
mydb.close()

在以上代码中，我们首先创建了一个MySQL连接，然后使用cursor对象执行SQL查询，将结果保存到变量data中。接着，我们创建一个dataframe对象，并将数据从data变量中读取到dataframe对象中。最后，我们使用ExcelWriter来导出数据并将其保存到excel文件中。因为我们知道结果只包含ID、Name和Age三列，因此我们手动指定了这三个列的列名。

这两个示例展示了如何使用pandas在python中实现导出excel文件的功能。您可以根据自己的需要调整这些示例中的代码，以满足您的实际需求。

Tags: ,