python写入数据到csv或xlsx文件的3种方法

下面将为您详细讲解Python如何写入数据到CSV或XLSX文件的三种方法。

一、CSV文件写入

1.1 方法一：使用csv库写入数据

import csv

# 自定义数据
data = [
    ['Jack', '27', 'Male'],
    ['Rose', '25', 'Female'],
    ['Tom', '30', 'Male']
]

# 写入CSV文件
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)

在代码中，首先引入了csv库，然后定义了一个自定义数据。接着，使用with open()语句打开一个CSV文件，指定newline=''参数以避免出现空行。

然后创建一个csv.writer对象，并使用循环将数据依次写入到文件中。最后，使用文件的write方法将数据写入到CSV文件中。

1.2 方法二：使用pandas库写入数据

import pandas as pd

# 自定义数据
data = {
    'name': ['Jack', 'Rose', 'Tom'],
    'age': ['27', '25', '30'],
    'gender': ['Male', 'Female', 'Male']
}

# 写入CSV文件
df = pd.DataFrame(data)
df.to_csv('data.csv', index=False)

在代码中，首先引入了pandas库，然后定义了一个自定义数据。接着，使用字典将数据存储到DataFrame中。

然后，将这些数据写入CSV文件，只需使用to_csv()方法，并将index设置为False。这样就可以将数据按照自定义的格式写入CSV文件中。

二、XLSX文件写入

2.1 方法一：使用openpyxl库写入数据

from openpyxl import Workbook

# 自定义数据
data = [
    ['Jack', '27', 'Male'],
    ['Rose', '25', 'Female'],
    ['Tom', '30', 'Male']
]

# 写入XLSX文件
wb = Workbook()
ws = wb.active
for row in data:
    ws.append(row)
wb.save('data.xlsx')

在代码中，首先引入了openpyxl库，然后定义了一个自定义数据。接下来，使用Workbook()方法创建一个新的Excel工作簿，并使用active属性获取当前的活动工作表。

然后，使用for循环将数据依次写入到工作表中。最后，使用save()方法将数据保存到XLSX文件中。

2.2 方法二：使用pandas库写入数据

import pandas as pd

# 自定义数据
data = {
    'name': ['Jack', 'Rose', 'Tom'],
    'age': ['27', '25', '30'],
    'gender': ['Male', 'Female', 'Male']
}

# 写入XLSX文件
df = pd.DataFrame(data)
df.to_excel('data.xlsx', index=False)

在代码中，首先引入了pandas库，然后定义了一个自定义数据。接下来，使用字典将数据存储到DataFrame中。

然后，使用to_excel()方法将DataFrame中的数据写入到XLSX文件中，并将index参数设置为False。这样就可以将数据按照自定义的格式写入XLSX文件中。

总的来说，Python写入CSV或XLSX文件的三种方法非常简单易懂。分别是使用csv库或pandas库来写入CSV文件，使用openpyxl库或pandas库来写入XLSX文件。它们有各自特点，可以根据实际情况选择适合自己的方式进行操作。