官方介绍: pandas的官方手册: https://pandas.pydata.org/pandas-docs/stable/ pandas官方读写数据文档: https://pandas.pydata.org/pandas-docs/stable/user_guide/io.html
一、使用pandas读取和写入csv文件
pandas.read_csv()语法: 1、使用pandas读取csv文件的全部数据:
- pd.read_csv("filepath",[encoding='编码'])
2、使用pandas读取csv文件的指定列方法:
- pd.read_csv("filepath",usecols=[0,1,2,...],[encoding='编码'])
3、使用pandas读取csv文件的指定行方法:
- pd.read_csv("filepath",[skiprows=n],nrows=m,[encoding='编码'])
pandas.read_csv(filepath_or_buffer, sep=', ', delimiter=None, header='infer', names=None, index_col=None, usecols=None, squeeze=False, prefix=None, mangle_dupe_cols=True, dtype=None, engine=None, converters=None, true_values=None, false_values=None, skipinitialspace=False, skiprows=None, nrows=None, na_values=None, keep_default_na=True, na_filter=True, verbose=False, skip_blank_lines=True, parse_dates=False, infer_datetime_format=False, keep_date_col=False, date_parser=None, dayfirst=False, iterator=False, chunksize=None, compression='infer', thousands=None, decimal=b'.', lineterminator=None, quotechar='"', quoting=0, escapechar=None, comment=None, encoding=None, dialect=None, tupleize_cols=None, error_bad_lines=True, warn_bad_lines=True, skipfooter=0, doublequote=True, delim_whitespace=False, low_memory=True, memory_map=False, float_precision=None)
常用参数解释:
- filepath_or_buffer #需要读取的文件及路径
- sep / delimiter 列分隔符,普通文本文件,应该都是使用结构化的方式来组织,才能使用dataframe
- header 文件中是否需要读取列名的一行,header=None(使用names自定义列名,否则默认0,1,2,...),header=0(将首行设为列名)
- names 如果header=None,那么names必须制定!否则就没有列的定义了。
- shkiprows= 10 # 跳过前十行
- nrows = 10 # 只去前10行
- usecols=[0,1,2,...] #需要读取的列,可以是列的位置编号,也可以是列的名称
- parse_dates = ['col_name'] # 指定某行读取为日期格式
- index_col = None /False /0,重新生成一列成为index值,0表示第一列,用作行索引的列编号或列名。可以是单个名称/数字或由多个名称/数字组成的列表(层次化索引)
- error_bad_lines = False # 当某行数据有问题时,不报错,直接跳过,处理脏数据时使用
- na_values = 'NULL' # 将NULL识别为空值
- encoding='utf-8' #指明读取文件的编码,默认utf-8
说明:① 读取csv/txt/tsv文件,返回一个DataFrame类型的对象。② 在读取的时候,默认会将第一行记录当成标题。如果没有标题,我们可以指定header=None。③ read_csv默认使用逗号作为分隔符,我们可以使用sep或delimiter来指定分隔符。④ 注意使用/修改为同一类型编码,否则会乱码。
# 案例:
import numpy as np
import pandas as pd
import warnings
warnings.filterwarnings("ignore")
print(np.__version__)
print(pd.__version__)
df = pd.read_csv(r"student.csv", header = None, sep=",",encoding="utf-8") #header = None,默认列名=0,1,2,3,...
df1 = pd.read_csv(r"student.csv", header = None,names=["id","name","sex","age","grade"], sep=",") #自定义列名
df2 = pd.read_csv(r"student.csv",header = None,index_col=None, sep=",") #默认自行生成行索引0,1,...
df3 = pd.read_csv(r"student.csv",header = None,index_col=[0], sep=",") #指定第1列作为行索引
df4 = pd.read_csv(r"student.csv",header=None,names=["id","name","sex","age","grade"],index_col=["name"], sep=",") #指定name列作为行索引
df5 = pd.read_csv(r"student.csv",header=None,usecols=[1,2,3],names=["name", "sex", "age"],index_col=0,sep=",") #指定列
display(df.head(3), df1.head(3), df2.head(3), df3.head(3), df4.head(3), df5.head(3))
1、使用pandas读取csv文件的全部数据:
pd.read_csv("filepath",[encoding='编码'])
如果存在编码(乱码)问题: (1)用记事本打开csv文件,另存为,编码格式改为utf-8然后用utf-8读取文件。 (2)用 csv编码的 “GB18030” 解码方式读取文件。 另外,由于python不支持中文,故一般在所有python代码开头第一行加上#coding=utf-8
In [1]: import numpy as np
In [2]: import pandas as pd
In [3]: df = pd.read_csv("E:\student.csv")
In [4]: print(df)
95002 刘晨 女 19 IS
0 95017 王风娟 女 18 IS
1 95018 王一 女 19 IS
...
20 95016 钱国 男 21 MA
In [5]: df = pd.read_csv("E:\student.csv",encoding='utf-8')
In [6]: print(df)
95002 刘晨 女 19 IS
0 95017 王风娟 女 18 IS
1 95018 王一 女 19 IS
...
20 95016 钱国 男 21 MA
细心观察数据,发现默认的他将我们的第一行数据当作是每个字段的标题啦。 (1)打开csv文件(为了避免编码问题,最好使用Notepad++打开编写修改),在第一行添加每列(columns)数据的标题。
In [1]: import numpy as np
In [2]: import pandas as pd
In [36]: df = pd.read_csv("e:\student.csv",encoding='utf-8')
In [37]: print(df)
编码 姓名 性别 年龄 班级
0 95002 刘晨 女 19 IS
1 95017 王风娟 女 18 IS
2 95018 王一 女 19 IS
仔细观察数据,很多时候我们需要的字段并不是那么多,现在我不想看到年龄的字段。
2、使用pandas读取csv文件的指定列方法:pd.read_csv("filepath",usecols=[0,1,2,...],[encoding='编码'])
In [1]: import numpy as np
In [2]: import pandas as pd
In [38]: df = pd.read_csv("e:\student.csv",usecols=[0,1,2,4],encoding='utf-8')
In [39]: print(df)
编码 姓名 性别 班级
0 95002 刘晨 女 IS
1 95017 王风娟 女 IS
In [1]: import numpy as np
In [2]: import pandas as pd
In [40]: df = pd.read_csv("e:\student.csv",usecols=['姓名','班级'],encoding='utf-8')
In [41]: print(df)
姓名 班级
0 刘晨 IS
1 王风娟 IS
2 王一 IS
...
3、使用pandas读取csv文件的指定行方法:pd.read_csv("filepath",[skiprows=n],nrows=m,[encoding='编码'])
pd.read_csv(路径,skiprows=n,nrows=m),忽略前n行,往下读m行
In [1]: import numpy as np
In [2]: import pandas as pd
In [47]: df = pd.read_csv("e:\student.csv",skiprows=5,nrows=2,encoding='utf-8')
In [48]: print(df)
95014 王小丽 女 19 CS
0 95019 邢小丽 女 19 IS
1 95020 赵钱 男 21 IS
二、使用pandas写入CSV文件
df.to_csv(path_or_buf,[sep=',' , na_rep=' ', ....])
写入csv文件是最常用的,csv文件默认用’,’作为分隔符。
df.to_csv(path_or_buf=None, sep=’,’, na_rep=”, float_format=None, columns=None, header=True, index=True, index_label=None, mode=’w’, encoding=None)
参数说明:
- path_or_buf:文件名、文件具体、相对路径、文件流等
- sep:文件分割符号,to_csv()的sep默认为’,’,可指定任意字符作为分隔符
- na_rep:将NaN转换为特定值。写入时NaN会被表示为空字符串,我们可能希望用其他值代替,如:‘- ’、‘/’、'NULL' 等
- columns:选择部分列写入。保留部分列且按列排序,columns=['B列列名','A列列名']
- header:忽略列名,header=None 不写入列名
- index:index=False 表示选择不写入索引
案例: pandas读取一个csv文件,将指定内容存入新的csv文件
In [1]: import numpy as np
In [2]: import pandas as pd
In [3]: df = pd.read_csv("e:\student.csv")
In [4]: print(df)
编码 姓名 性别 年龄 班级
0 95002 刘晨 女 19 IS
1 95017 王风娟 女 18 IS
2 95018 王一 女 19 IS
In [5]: df.to_csv("e:\student01.csv",sep=",",columns=['姓名','班级'],index=True)
In [6]: df = pd.read_csv('e:\student01.csv')
In [7]: print(df)
Unnamed: 0 姓名 班级
0 0 刘晨 IS
1 1 王风娟 IS
2 2 王一 IS