python写入csv文件的方法有哪些

2024-04-23

在Python中,写入CSV文件的方法有以下几种:

  1. 使用csv模块:
import csv

data = [['Name', 'Age', 'Country'],
        ['Alice', 30, 'USA'],
        ['Bob', 25, 'Canada'],
        ['Cathy', 35, 'UK']]

with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)
  1. 使用pandas库:
import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Cathy'],
        'Age': [30, 25, 35],
        'Country': ['USA', 'Canada', 'UK']}

df = pd.DataFrame(data)
df.to_csv('data.csv', index=False)
  1. 使用numpy库:
import numpy as np

data = np.array([['Name', 'Age', 'Country'],
                 ['Alice', 30, 'USA'],
                 ['Bob', 25, 'Canada'],
                 ['Cathy', 35, 'UK']])

np.savetxt('data.csv', data, delimiter=',', fmt='%s')

以上是几种常用的写入CSV文件的方法,可以根据具体情况选择适合自己的方法。