欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

程序员文章站 2023-08-23 12:55:28
import pandas as pdimport numpy as np查看Pandas版本pd.__version__一、文件读取与写入1.读取(a)csv格式df = pd.read_csv(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\table.csv')df.head()(b)txt格式df_txt = pd.read_table(r'C:\Users\cheny...

import pandas as pd
import numpy as np

查看Pandas版本

pd.__version__

一、文件读取与写入

1.读取

(a)csv格式

df = pd.read_csv(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\table.csv')
df.head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

(b)txt格式

df_txt = pd.read_table(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\table.txt') #可设置sep分隔符参数
df_txt

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

(c)xls或xlsx格式

#需要安装xlrd包
df_excel = pd.read_excel(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\table.xlsx')
df_excel.head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

2. 写入

(a)csv格式

df.to_csv('data/new_table.csv')
#df.to_csv('data/new_table.csv', index=False) #保存时除去行索引

(b)xls或xlsx格式

#需要安装openpyxl
df.to_excel('data/new_table2.xlsx', sheet_name='Sheet1')

二、基本数据结构

1.Series

(a)创建一个Series

对于一个Series,其中最常用的属性为值(values),索引(index),名字(name),类型(dtype)

s = pd.Series(np.random.randn(5),index=['a','b','c','d','e'],name='这是一个Series',dtype='float64')
s
a    1.456004
b   -0.707533
c    0.485580
d   -1.356148
e   -0.202208
Name: 这是一个Series, dtype: float64

(b)访问Series属性

s.values

array([ 1.45600438, -0.70753348, 0.48558013, -1.35614791, -0.20220802])

s.name

‘这是一个Series’

s.index

Index([‘a’, ‘b’, ‘c’, ‘d’, ‘e’], dtype=‘object’)

s.dtype

dtype(‘float64’)

(c)取出某一个元素

s['a']

1.4560043774073959

(d)调用方法

s.mean()

-0.06486097947792556

Series有相当多的方法可以调用:

print([attr for attr in dir(s) if not attr.startswith('_')])
['T', 'a', 'abs', 'add', 'add_prefix', 'add_suffix', 'agg', 'aggregate', 'align', 'all', 'any', 'append', 'apply', 'argmax', 'argmin', 'argsort', 'array', 'asfreq', 'asof', 'astype', 'at', 'at_time', 'attrs', 'autocorr', 'axes', 'b', 'between', 'between_time', 'bfill', 'bool', 'c', 'clip', 'combine', 'combine_first', 'convert_dtypes', 'copy', 'corr', 'count', 'cov', 'cummax', 'cummin', 'cumprod', 'cumsum', 'd', 'describe', 'diff', 'div', 'divide', 'divmod', 'dot', 'drop', 'drop_duplicates', 'droplevel', 'dropna', 'dtype', 'dtypes', 'duplicated', 'e', 'empty', 'eq', 'equals', 'ewm', 'expanding', 'explode', 'factorize', 'ffill', 'fillna', 'filter', 'first', 'first_valid_index', 'floordiv', 'ge', 'get', 'groupby', 'gt', 'hasnans', 'head', 'hist', 'iat', 'idxmax', 'idxmin', 'iloc', 'index', 'infer_objects', 'interpolate', 'is_monotonic', 'is_monotonic_decreasing', 'is_monotonic_increasing', 'is_unique', 'isin', 'isna', 'isnull', 'item', 'items', 'iteritems', 'keys', 'kurt', 'kurtosis', 'last', 'last_valid_index', 'le', 'loc', 'lt', 'mad', 'map', 'mask', 'max', 'mean', 'median', 'memory_usage', 'min', 'mod', 'mode', 'mul', 'multiply', 'name', 'nbytes', 'ndim', 'ne', 'nlargest', 'notna', 'notnull', 'nsmallest', 'nunique', 'pct_change', 'pipe', 'plot', 'pop', 'pow', 'prod', 'product', 'quantile', 'radd', 'rank', 'ravel', 'rdiv', 'rdivmod', 'reindex', 'reindex_like', 'rename', 'rename_axis', 'reorder_levels', 'repeat', 'replace', 'resample', 'reset_index', 'rfloordiv', 'rmod', 'rmul', 'rolling', 'round', 'rpow', 'rsub', 'rtruediv', 'sample', 'searchsorted', 'sem', 'set_axis', 'shape', 'shift', 'size', 'skew', 'slice_shift', 'sort_index', 'sort_values', 'squeeze', 'std', 'sub', 'subtract', 'sum', 'swapaxes', 'swaplevel', 'tail', 'take', 'to_clipboard', 'to_csv', 'to_dict', 'to_excel', 'to_frame', 'to_hdf', 'to_json', 'to_latex', 'to_list', 'to_markdown', 'to_numpy', 'to_period', 'to_pickle', 'to_sql', 'to_string', 'to_timestamp', 'to_xarray', 'transform', 'transpose', 'truediv', 'truncate', 'tshift', 'tz_convert', 'tz_localize', 'unique', 'unstack', 'update', 'value_counts', 'values', 'var', 'view', 'where', 'xs']

2.DataFrame

(a)创建一个DataFrame

df = pd.DataFrame({'col1':list('abcde'),'col2':range(5,10),'col3':[1.3,2.5,3.6,4.6,5.8]},
                 index=list('一二三四五'))
df

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

(b)从DataFrame取出一列为Series

df['col1']
1.32.53.64.65.8
Name: col3, dtype: float64
type(df)

pandas.core.frame.DataFrame

type(df['col3'])

pandas.core.series.Series

(c)修改行或列名

df.rename(index={'一':'1'},columns={'col1':'第一列'})

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

(d)调用属性和方法

df.index

Index([‘一’, ‘二’, ‘三’, ‘四’, ‘五’], dtype=‘object’)

df.columns

Index([‘col1’, ‘col2’, ‘col3’], dtype=‘object’)

df.values

array([[‘a’, 5, 1.3],
[‘b’, 6, 2.5],
[‘c’, 7, 3.6],
[‘d’, 8, 4.6],
[‘e’, 9, 5.8]], dtype=object)

df.shape

(5, 3)

df.mean() #本质上是一种Aggregation操作,将在第3章详细介绍

col2 7.00
col3 3.56
dtype: float64

(e)索引对齐特性

这是Pandas中非常强大的特性,不理解这一特性有时就会造成一些麻烦

df1 = pd.DataFrame({'A':[1,2,3]},index=[1,2,3])
df2 = pd.DataFrame({'A':[1,2,3]},index=[3,1,2])
df1-df2 #由于索引对齐,因此结果不是0

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

(f)列的删除与添加

对于删除而言,可以使用drop函数或del或pop

df.drop(index='五',columns='col1') #设置inplace=True后会直接在原DataFrame中改动

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

df['col1']=[1,2,3,4,5]
del df['col1']
df

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
pop方法直接在原来的DataFrame上操作,且返回被删除的列,与python中的pop函数类似

df['col1']=[1,2,3,4,5]
df.pop('col1')

一 1
二 2
三 3
四 4
五 5
Name: col1, dtype: int64

df

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
可以直接增加新的列,也可以使用assign方法

df1['B']=list('abc')
df1

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

df1.assign(C=pd.Series(list('def')))
#思考:为什么会出现NaN?(提示:索引对齐)assign左右两边的索引不一样,请问结果的索引谁说了算?

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
但assign方法不会对原DataFrame做修改

(g)根据类型选择列

df.select_dtypes(include=['number']).head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

df.select_dtypes(include=['float']).head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

(h)将Series转换为DataFrame

# 此部操作之后,s成为一series
s = df.mean()
s.name='to_DataFrame'
s
col2    7.00
col3    3.56
Name: to_DataFrame, dtype: float64
# 将s,转换成dataframe
s.to_frame()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
使用T符号可以转置

s.to_frame().T

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

三、常用基本函数

df = pd.read_csv(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\table.csv')

从下面开始,包括后面所有章节,我们都会用到这份虚拟的数据集

1.head和tail

df.head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

df.tail()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
可以指定n参数显示多少行

df.head(2)

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

2.unique和nunique

nunique显示有多少个唯一值

df['Physics'].nunique()

7
unique显示所有的唯一值

df['Physics'].unique()
array(['A+', 'B+', 'B-', 'A-', 'B', 'A', 'C'], dtype=object)

3.count和value_counts

count返回非缺失值元素个数

df['Physics'].count()

35
value_counts返回每个元素有多少个

df['Physics'].value_counts()
B+    9
B     8
B-    6
A     4
A-    3
A+    3
C     2
Name: Physics, dtype: int64

4.describe和info

info函数返回有哪些列、有多少非缺失值、每列的类型

df.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 35 entries, 0 to 34
Data columns (total 9 columns):
 #   Column   Non-Null Count  Dtype  
---  ------   --------------  -----  
 0   School   35 non-null     object 
 1   Class    35 non-null     object 
 2   ID       35 non-null     int64  
 3   Gender   35 non-null     object 
 4   Address  35 non-null     object 
 5   Height   35 non-null     int64  
 6   Weight   35 non-null     int64  
 7   Math     35 non-null     float64
 8   Physics  35 non-null     object 
dtypes: float64(1), int64(3), object(5)
memory usage: 2.6+ KB

describe默认统计数值型数据的各个统计量

df.describe()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
可以自行选择分位数

df.describe(percentiles=[.05, .25, .75, .95])

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
对于非数值型也可以用describe函数

df['Physics'].describe()
count     35
unique     7
top       B+
freq       9
Name: Physics, dtype: object

5.idxmax和nlargest

idxmax函数返回最大值所在索引,在某些情况下特别适用,idxmin功能类似

df['Math'].idxmax()

5

nlargest函数返回前几个大的元素值,nsmallest功能类似

df['Math'].nlargest(3)
5     97.0
28    95.5
11    87.7
Name: Math, dtype: float64

6.clip和replace

clip和replace是两类替换函数

clip是对超过或者低于某些值的数进行截断

df['Math'].head()
0    34.0
1    32.5
2    87.2
3    80.4
4    84.8
Name: Math, dtype: float64
df['Math'].clip(33,80).head()
0    34.0
1    33.0
2    80.0
3    80.0
4    80.0
Name: Math, dtype: float64
df['Math'].mad()

16.924244897959188

replace是对某些值进行替换

df['Address'].head()
0    street_1
1    street_2
2    street_2
3    street_2
4    street_4
Name: Address, dtype: object
df['Address'].replace(['street_1','street_2'],['one','two']).head()
0         one
1         two
2         two
3         two
4    street_4
Name: Address, dtype: object

通过字典,可以直接在表中修改

df.replace({'Address':{'street_1':'one','street_2':'two'}}).head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

7.apply函数

apply是一个*度很高的函数,在第3章我们还要提到

对于Series,它可以迭代每一列的值操作:

df['Math'].apply(lambda x:str(x)+'!').head() #可以使用lambda表达式,也可以使用函数
0    34.0!
1    32.5!
2    87.2!
3    80.4!
4    84.8!
Name: Math, dtype: object

对于DataFrame,它在默认axis=0下可以迭代每一个列操作:

df.apply(lambda x:x.apply(lambda x:str(x)+'!')).head() #这是一个稍显复杂的例子,有利于理解apply的功能

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
对于Pandas中axis参数的理解如下:

https://blog.csdn.net/qq_40317204/article/details/106993293

四、排序

1.索引排序

df.set_index('Math').head() #set_index函数可以设置索引,将在下一章详细介绍

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

df.set_index('Math').sort_index().head() #可以设置ascending参数,默认为升序,True

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

2.值排序

df.sort_values(by='Class').head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)
多个值排序,即先对第一层排,在第一层相同的情况下对第二层排序

df.sort_values(by=['Address','Height']).head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

五、问题与练习

1. 问题

使用assign添加列的时候,为什么会出现NaN?(提示:索引对齐)assign左右两边的索引不一样,请问结果的索引谁说了算?(内容定位:二-1-f)

2. 练习

【练习一】 现有一份关于美剧《权力的游戏》剧本的数据集,请解决以下问题:
(a)在所有的数据中,一共出现了多少人物?
(b)以单元格计数(即简单把一个单元格视作一句),谁说了最多的话?
(c)以单词计数,谁说了最多的单词?(不是单句单词最多,是指每人说过单词的总数最多,为了简便,只以空格为单词分界点,不考虑其他情况)

(a)

pd.read_csv(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\Game_of_Thrones_Script.csv').head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

df['Name'].nunique()

564

(b)

df['Name'].value_counts()

可以得到每个人物的台词数量:

tyrion lannister          1760
jon snow                  1133
daenerys targaryen        1048
cersei lannister          1005
jaime lannister            945
                          ... 
night watch stable boy       1
tell me something            1
little sam                   1
ser vance                    1
allister                     1
Name: Name, Length: 564, dtype: int64
df['Name'].value_counts().index[0]

台词最多的人物为:

'tyrion lannister'

(c)

df_words = df.assign(Words=df['Sentence'].apply(lambda x:len(x.split()))).sort_values(by='Name')
df_words.head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

L_count = []
N_words = list(zip(df_words['Name'],df_words['Words']))
for i in N_words:
    if i == N_words[0]:
        L_count.append(i[1])
        last = i[0]
    else:
        L_count.append(L_count[-1]+i[1] if i[0]==last else i[1])
        last = i[0]
df_words['Count']=L_count
df_words['Name'][df_words['Count'].idxmax()]

得到说过单词总数最多的人:

'tyrion lannister'

**【练习二】**现有一份关于科比的投篮数据集,请解决如下问题:
(a)哪种action_type和combined_shot_type的组合是最多的?
(a)在所有被记录的game_id中,遭遇到最多的opponent是一个支?(由于一场比赛会有许多次投篮,但对阵的对手只有一个,本题相当于问科比和哪个队交锋次数最多)

(a)

df = pd.read_csv(r'C:\Users\chenyiqun\Desktop\joyful-pandas-master\joyful-pandas-master\data\Kobe_data.csv',index_col='shot_id')
df.head()

Pandas基础(文件读取与写入、Series和Dataframe、常用基本函数、排序)

pd.Series(list(zip(df['action_type'],df['combined_shot_type']))).value_counts()

得到所有的组合及数量:

(Jump Shot, Jump Shot)                             18880
(Layup Shot, Layup)                                 2567
(Driving Layup Shot, Layup)                         1978
(Turnaround Jump Shot, Jump Shot)                   1057
(Fadeaway Jump Shot, Jump Shot)                     1048
(Running Jump Shot, Jump Shot)                       926
(Pullup Jump shot, Jump Shot)                        476
(Turnaround Fadeaway shot, Jump Shot)                439
(Slam Dunk Shot, Dunk)                               411
(Reverse Layup Shot, Layup)                          395
(Jump Bank Shot, Jump Shot)                          333
(Driving Dunk Shot, Dunk)                            310
(Dunk Shot, Dunk)                                    262
(Tip Shot, Tip Shot)                                 182
(Alley Oop Dunk Shot, Dunk)                          122
(Step Back Jump shot, Jump Shot)                     118
(Floating Jump shot, Jump Shot)                      114
(Driving Reverse Layup Shot, Layup)                   97
(Hook Shot, Hook Shot)                                84
(Driving Finger Roll Shot, Layup)                     82
(Alley Oop Layup shot, Layup)                         80
(Reverse Dunk Shot, Dunk)                             75
(Running Layup Shot, Layup)                           72
(Turnaround Bank shot, Bank Shot)                     71
(Driving Finger Roll Layup Shot, Layup)               69
(Driving Slam Dunk Shot, Dunk)                        48
(Running Bank shot, Bank Shot)                        48
(Running Hook Shot, Hook Shot)                        41
(Finger Roll Layup Shot, Layup)                       33
(Fadeaway Bank shot, Jump Shot)                       31
(Finger Roll Shot, Layup)                             28
(Driving Jump shot, Jump Shot)                        28
(Jump Hook Shot, Jump Shot)                           24
(Running Dunk Shot, Dunk)                             19
(Reverse Slam Dunk Shot, Dunk)                        16
(Putback Layup Shot, Layup)                           15
(Follow Up Dunk Shot, Dunk)                           15
(Driving Hook Shot, Hook Shot)                        14
(Turnaround Hook Shot, Hook Shot)                     14
(Pullup Bank shot, Bank Shot)                         12
(Running Reverse Layup Shot, Layup)                   11
(Cutting Layup Shot, Layup)                            6
(Running Finger Roll Layup Shot, Layup)                6
(Driving Bank shot, Bank Shot)                         5
(Hook Bank Shot, Bank Shot)                            5
(Putback Dunk Shot, Dunk)                              5
(Driving Floating Jump Shot, Jump Shot)                5
(Running Finger Roll Shot, Layup)                      4
(Running Pull-Up Jump Shot, Jump Shot)                 4
(Tip Layup Shot, Layup)                                2
(Running Tip Shot, Tip Shot)                           2
(Putback Slam Dunk Shot, Dunk)                         2
(Turnaround Finger Roll Shot, Layup)                   2
(Cutting Finger Roll Layup Shot, Layup)                1
(Driving Floating Bank Jump Shot, Jump Shot)           1
(Turnaround Fadeaway Bank Jump Shot, Jump Shot)        1
(Running Slam Dunk Shot, Dunk)                         1
dtype: int64
pd.Series(list(zip(df['action_type'],df['combined_shot_type']))).value_counts().index[0]

(‘Jump Shot’, ‘Jump Shot’)

(b)

pd.Series(list(list(zip(*(pd.Series(list(zip(df['game_id'],df['opponent'])))
                          .unique()).tolist()))[1])).value_counts()

得到与所有对手交战次数:

SAS    91
PHX    87
UTA    84
DEN    83
POR    81
SAC    80
HOU    77
MIN    76
LAC    68
GSW    67
DAL    64
MEM    49
BOS    44
SEA    44
IND    39
DET    38
PHI    37
ORL    34
NYK    34
CLE    34
OKC    33
TOR    33
WAS    32
MIA    32
MIL    31
CHA    31
NOH    31
CHI    31
ATL    29
NJN    28
VAN    18
NOP    16
BKN     3
dtype: int64
​
pd.Series(list(list(zip(*(pd.Series(list(zip(df['game_id'],df['opponent'])))
                          .unique()).tolist()))[1])).value_counts().index[0]

‘SAS’

部分python函数参考文章:
Python zip() 函数解析
Python split()函数解析

Pandas基础:
https://github.com/datawhalechina/joyful-pandas/blob/master/%E7%AC%AC1%E7%AB%A0%20Pandas%E5%9F%BA%E7%A1%80.ipynb

练习题参考答案:
https://github.com/datawhalechina/joyful-pandas/blob/master/%E5%8F%82%E8%80%83%E7%AD%94%E6%A1%88.ipynb

所用数据集下载连接:
https://download.csdn.net/download/qq_40317204/12568932

本文地址:https://blog.csdn.net/qq_40317204/article/details/106979983