一、loc和iloc函数
- loc函数:通过行索引 “index” (或者列索引"columns")中的具体值来获取行或列的数据
- iloc函数:通过行号(或者列号)来取行或列的数据
二、实例
- 利用loc、iloc函数提取行数据
import numpy as np
import pandas as pd
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
'''
print(data.loc['s1'])
print(data.iloc[0])
'''
a1 0
a2 1
a3 2
a4 3
'''
- 利用loc、iloc函数提取列数据
print(data.loc[:,['a1']])
print(data.iloc[:,[0]])
'''
a1
s1 0
s2 4
s3 8
s4 12
'''
print(data.loc[:,['a1', 'a3']])
print(data.iloc[:,[0,2]])
'''
a1 a3
s1 0 2
s2 4 6
s3 8 10
s4 12 14
'''
- 利用loc、iloc函数提取指定行、列数据
print(data.loc[['s1', 's2'],['a1', 'a2']])
print(data.iloc[[0,1],[0,1]])
'''
a1 a2
s1 0 1
s2 4 5
'''
- 利用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
'''
- 利用loc函数,根据某个数据来提取数据所在的行
print(data.loc[data['a1']==0])
'''
a1 a2 a3 a4
s1 0 1 2 3
'''