当前位置: 代码迷 >> 综合 >> Python dataframe 行 元素 索引方法
  详细解决方案

Python dataframe 行 元素 索引方法

热度:71   发布时间:2023-12-11 17:18:13.0

在dataframe中根据一定的条件,得到符合要求的某行元素所在的位置。

代码如下所示:


df = pd.DataFrame({
   'BoolCol': [1, 2, 3, 3, 4],'attr': [22, 33, 22, 44, 66]},index=[10,20,30,40,50])
print(df)
a = df[(df.BoolCol==3)&(df.attr==22)].index.tolist()
print(a)
df如下所示,以上通过选取“BoolCol”取值为3且“attr”取值为22的行,得到该行在df中的位置
注意:返回的位置为index列表,根据index的不同而不同,这点易于数组中默认的下标。BoolCol  attr
10        1    22
20        2    33
30        3    22
40        3    44
50        4    66
[30]

更多量化学习资源

介绍.png
扫上方二维码,关注公众账号 量化投资学院 ,获取下列免费资源
- 回复“热点研报”,获取近年热点券商金融工程研究报告
- 回复“Python3”,获取Python免费学习教程
* 更多福利请点击此链接

关注StudyQuant

  • 课程
    –量化投资与数据分析实战
    –量化投资与数字货币实战
    –免费Python及量化投资公开课

  • 知乎

  • 简书
  • 公众号