Pandas中的loc和iloc函数

一、loc和iloc函数

• loc函数：通过行索引 “index” （或者列索引"columns"）中的具体值来获取行或列的数据
• iloc函数：通过行号（或者列号）来取行或列的数据

二、实例

1. 利用loc、iloc函数提取行数据

import numpy as np
import pandas as pd
#创建一个4x4的表格
data=pd.DataFrame(np.arange(16).reshape(4,4),index=['s1', 's2', 's3', 's4'],columns=['a1', 'a2', 'a3', 'a4'])
print(data)
'''
    a1  a2  a3  a4
s1   0   1   2   3
s2   4   5   6   7
s3   8   9  10  11
s4  12  13  14  15
'''
#行索引为's1'的数据
print(data.loc['s1'])
print(data.iloc[0])  #取第一行数据，索引为's1'的行就是第一行，所以结果相同
'''
a1    0
a2    1
a3    2
a4    3
'''

2. 利用loc、iloc函数提取列数据

#列索引为'a1'的数据
print(data.loc[:,['a1']])
print(data.iloc[:,[0]])  #取第一列数据，索引为'a1'的列就是第一列，所以结果相同
'''
    a1
s1   0
s2   4
s3   8
s4  12
'''

#列索引为'a1'、'a3'的数据
print(data.loc[:,['a1', 'a3']])
print(data.iloc[:,[0,2]])
'''
    a1  a3
s1   0   2
s2   4   6
s3   8  10
s4  12  14
'''

3. 利用loc、iloc函数提取指定行、列数据

print(data.loc[['s1', 's2'],['a1', 'a2']])
print(data.iloc[[0,1],[0,1]])
'''
    a1  a2
s1   0   1
s2   4   5
'''

4. 利用loc、iloc函数提取所有数据

print(data.loc[:,:])
print(data.iloc[:,:])
'''
    a1  a2  a3  a4
s1   0   1   2   3
s2   4   5   6   7
s3   8   9  10  11
s4  12  13  14  15
'''

5. 利用loc函数，根据某个数据来提取数据所在的行

print(data.loc[data['a1']==0]) #提取data数据(筛选条件: 'a1'列中数字为0所在的行数据)
'''
    a1  a2  a3  a4
s1   0   1   2   3
'''