Pandas库中的loc和iloc函数是用于对数据进行定位和提取的两个常用函数。
loc函数
loc函数是基于标签的索引,它的使用方法如下:
# 对DataFrame进行索引 # 获取'A'列的所有数据 df.loc[:,'A'] # 获取'A','B'列的所有数据 df.loc[:,['A','B']] # 获取'A'列中索引值为1的数据 df.loc[1,'A'] # 获取'A','B'列中索引值为1的数据 df.loc[1,['A','B']] # 获取'A'列中索引值为1,2,3的数据 df.loc[[1,2,3],'A'] # 获取'A','B'列中索引值为1,2,3的数据 df.loc[[1,2,3],['A','B']]
iloc函数
iloc函数是基于整数的索引,它的使用方法如下:
# 对DataFrame进行索引 # 获取第一列的所有数据 df.iloc[:,0] # 获取第一列和第二列的所有数据 df.iloc[:,[0,1]] # 获取第一列中索引值为1的数据 df.iloc[1,0] # 获取第一列和第二列中索引值为1的数据 df.iloc[1,[0,1]] # 获取第一列中索引值为1,2,3的数据 df.iloc[[1,2,3],0] # 获取第一列和第二列中索引值为1,2,3的数据 df.iloc[[1,2,3],[0,1]]
loc和iloc函数的使用方式比较简单,只要掌握它们的使用方法,就可以很方便的对数据进行定位和提取。