栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

Python Pandas:获取列匹配特定值的行的索引

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Python Pandas:获取列匹配特定值的行的索引

df.iloc[i]
返回的
ith
df
i
不引用索引标签,
i
是基于0的索引。

相反, 该属性

index
返回实际的索引标签,而不是数字的行索引:

df.index[df['BoolCol'] == True].tolist()

或等效地,

df.index[df['BoolCol']].tolist()

通过使用具有非默认索引且不等于行的数字位置的Dataframe,您可以非常清楚地看到差异:

df = pd.Dataframe({'BoolCol': [True, False, False, True, True]},       index=[10,20,30,40,50])In [53]: dfOut[53]:    BoolCol10    True20   False30   False40    True50    True[5 rows x 1 columns]In [54]: df.index[df['BoolCol']].tolist()Out[54]: [10, 40, 50]

如果要使用索引

In [56]: idx = df.index[df['BoolCol']]In [57]: idxOut[57]: Int64Index([10, 40, 50], dtype='int64')

那么您可以使用

loc
代替来选择行
iloc

In [58]: df.loc[idx]Out[58]:    BoolCol10    True40    True50    True[3 rows x 1 columns]

注意,

loc
也可以接受布尔数组

In [55]: df.loc[df['BoolCol']]Out[55]:    BoolCol10    True40    True50    True[3 rows x 1 columns]

如果您有一个布尔数组,

mask
并且需要序数索引值,则可以使用进行计算
np.flatnonzero

In [110]: np.flatnonzero(df['BoolCol'])Out[112]: array([0, 3, 4])

用于

df.iloc
按顺序索引选择行:

In [113]: df.iloc[np.flatnonzero(df['BoolCol'])]Out[113]:    BoolCol10    True40    True50    True


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/662477.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号