一、用途與特點(diǎn)
DataFrame是Pandas中最重要的數(shù)據(jù)結(jié)構(gòu)之一,而DataFrame.loc則是對(duì)于DataFrame進(jìn)行行列選取的一種方法。它通過行標(biāo)簽名和列標(biāo)簽名進(jìn)行數(shù)據(jù)選取。
相對(duì)于其他方法,DataFrame.loc具有更高的靈活性和準(zhǔn)確性。與iloc根據(jù)行列的位置來進(jìn)行選取不同,使用loc可以根據(jù)行列的標(biāo)簽名稱來進(jìn)行選取。并且在同時(shí)進(jìn)行行列選取時(shí),使用loc的效率更高。
# 用于基于標(biāo)簽的索引(即行列標(biāo)簽)進(jìn)行選擇
# df.loc[row, column]
# row可以是行標(biāo)簽也可以是布爾數(shù)組,column可以是列標(biāo)簽也可以是布爾數(shù)組
二、基本用法
選擇某些列:
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randn(6,4), columns=list('ABCD'), index=list('abcdef'))
# 根據(jù)標(biāo)簽獲取列數(shù)據(jù)
print(df.loc[:, ['A', 'B']])
選擇某些行:
# 根據(jù)標(biāo)簽獲取行數(shù)據(jù)
print(df.loc[['a', 'b', 'f'], :])
選擇某個(gè)區(qū)域的數(shù)據(jù):
# 根據(jù)標(biāo)簽獲取行和列的區(qū)域數(shù)據(jù)
print(df.loc['d':'f', 'A':'C'])
三、高級(jí)用法
布爾索引
在DataFrame.loc中使用布爾值可以進(jìn)行高級(jí)篩選。
# 布爾索引
df.loc[df['A'] > 0, ['A','B']]
使用函數(shù)進(jìn)行映射
在DataFrame中使用applymap或apply方法可以對(duì)數(shù)據(jù)進(jìn)行轉(zhuǎn)換。在.loc后使用,我們可以完成數(shù)據(jù)篩選和轉(zhuǎn)換的一步操作。
# 對(duì)篩選出來的數(shù)據(jù)進(jìn)行applymap的map函數(shù)映射操作
df.loc[df['A'] > 0, ['A','B']].applymap(lambda x: x*2)
使用where方法
使用where方法進(jìn)行條件篩選。對(duì)于篩選為False的數(shù)據(jù),其對(duì)應(yīng)位置上的數(shù)據(jù)會(huì)被填充為NaN。
# 對(duì)q值保留填充,而對(duì)于其他的行列標(biāo)簽均標(biāo)記為NaN
df.loc[df['A'] > 0, 'B'] = np.nan
df.where(pd.notna(df), df.mean(), axis='columns')
四、總結(jié)
本文基于DataFrame的高級(jí)索引方法DataFrame.loc進(jìn)行了詳細(xì)講解,介紹了其基本使用方法和高級(jí)應(yīng)用方法。
對(duì)于Pandas用戶而言,熟悉DataFrame.loc的使用方法可以幫助我們更加靈活地對(duì)數(shù)據(jù)進(jìn)行篩選和轉(zhuǎn)換。它是我們使用Pandas解決數(shù)據(jù)分析和處理問題的重要方法。