使用Python读写csv文件

# -*- coding: UTF-8 -*-
import csv
file = "demo.csv"
with open(file, "r")as f:
    f_csv = csv.reader(f)
    for row in f_csv:
        print(row)

import pandas as pd

file = "demo.csv"
data = pd.read_csv(file, encoding="gbk")
print(data.head(2))  # 打印前2行
print("*" * 100)
print(data.columns)  # 返回全部列名
print("*" * 100)
print(data.shape)  # 返回csv文件形状  例如(4, 3) => 4行3列
print("*" * 100)
print(data.loc[1:2])  # 打印第1到2行

# -*- coding: UTF-8 -*-
import csv

csvFile = open("test.csv", "w")
writer = csv.writer(csvFile)

# 写入列名。
writer.writerow(["id", "name", "age"])

row1 = ["1001", "Tom", 30]
row2 = ["1002", "David", 25]
row3 = ["1003", "Lily", 20]
row4 = ["1004", "韩梅梅", 15]

writer.writerows([row1, row2, row3, row4])
csvFile.close()

# -*- coding: UTF-8 -*-
import csv

import csv

headers = ['id', 'name', 'age']

row1 = {"id": "1001", "name": "Tom", "age": 30}
row2 = {"id": "1002", "name": "Lily", "age": 35}
row3 = {"id": "1003", "name": "李雷", "age": 40}
row4 = {"id": "1004", "name": "韩梅梅", "age": 45}

rows = [row1, row2, row3, row4]

with open('test2.csv', 'w', newline='')as f:
    f_csv = csv.DictWriter(f, headers)
    f_csv.writeheader()
    f_csv.writerows(rows)