开发者社区 问答 正文

如何获得dataframe行的和大于100的最末n行索引?

如何获得dataframe行的和大于100的最末n行索引?

展开
收起
游客y244y7ln2rlpa 2021-12-05 20:24:17 398 分享 版权
1 条回答
写回答
取消 提交回答
  • df = pd.DataFrame(np.random.randint(10, 40, 16).reshape(-1, 4))
    print(df)
    # dataframe每行的和
    rowsums = df.apply(np.sum, axis=1)
    
    # 选取大于100的最末两行索引
    # last_two_rows = df.iloc[np.where(rowsums > 100)[0][-2:], :]
    nline = np.where(rowsums > 100)[0][-2:]
    nline
    
    #>	    0   1   2   3
    	0  19  34  15  12
    	1  38  35  14  26
    	2  39  32  18  20
    	3  28  27  36  38
    
    #>	array([2, 3], dtype=int64)
    
    
    2021-12-05 22:22:57
    赞同 展开评论
问答分类:
问答地址: