csv文件的操作

一: 用csv包, 操作csv文件

import csv

# 提取数据

with open('csv文件名称', '打开方式') as f:

       # 使用f文件句柄, 进行读取

        rows = csv.reader(f)

       # for row in rows:

                print(row)

# 写入数据

with open('csv文件名称', '打开方式') as f:

       # 创建writer

        writer = csv.writer(f)

        writer.writerow('写入的内容')

二: pandas读取csv文件

import pandas as pd

# 读取数据, 读取到的是一个df对象

df = pd.read_csv('csv文件名称')

# 预览前五条数据

df.head()

# 预览后五条数据

df.tail()

# 获取某一列的数据, 获取到的是一个Series对象

df['某列']

# 获取指定多列的数据, 是一个DataFrame

df[['A', 'C']]

# 获取指定列的所有值, 返回一个ndarray二维数组

df[['A', 'C']].values

# 获取指定列的列索引

df[['A', 'C']].keys()

三. 将df对象保存为csv文件

# index设置为None, 不会将索引写入文件

df.to_csv('csv文件名', encoding='utf-8', index=None)