python pandas df.loc[]的典型用法mp.weixin.qq.com
pandas中的df.loc[]主要是根据DataFrame的行标和列标进行数据的筛选的,如下图红框部分所示:
其接受两个参数:行标和列标,当列标省略时,默认获取整行数据。两个参数都可以以字符,切片以及列表的形式传入。
>>> import pandas as pd
>>> import numpy as np
>>> df = pd.read_excel(r'D:/myExcel/1.xlsx', index_col='id')
>>> df
name score grade
id
a bog 45 A
c jiken 67 B
d bob 23 A
b jiken 34 B
f lucy 98 A
e tidy 75 B
# 指定行标'a'以及'f'行,第二个参数省略,则获取整行
>>> df.loc[['a', 'f']]
name score grade
id
a bog 45 A
f lucy 98 A
# 以切片形式传入,会按照df中的顺序获取
# 此处需要注意,并不是只获取'a', 'b'行