为了账号安全,请及时绑定邮箱和手机立即绑定

按列索引获取值,其中行是特定值

按列索引获取值,其中行是特定值

catspeake 2021-08-17 16:15:51
我有一个数据框 sheet_overview:    Unnamed: 0 Headline Unnamed: 2 Unnamed: 3  0         nan    1. 1.   username       Erik  1         nan    1. 2.    userage         23  2         nan    1. 3.   favorite        ice我想通过在第二列中查找“ 1. 2. ”来获得值23。如果我不想进入列名,我必须使用索引。我的问题是,如果我的方法太复杂了。 它有效,但似乎太多了,而且不是很pythonic:age = sheet_overview.iloc[                    sheet_overview[                        sheet_overview.iloc[:, 1] == '1. 2.']                        .index[0], 3]
查看完整描述

1 回答

?
一只名叫tom的猫

TA贡献1906条经验 获得超3个赞

添加values用于过滤器的 numpy 数组iloc,然后next用于返回第一个匹配的值 - 如果不存在获取missing:


a = sheet_overview.iloc[(sheet_overview.iloc[:, 1] == '1. 2.').values, 3]


a = next(iter(a), 'missing')

print (a)

23

如果性能很重要,请使用numba:


from numba import njit


@njit

def first_val(A, k):

    a = A[:, 0]

    b = A[:, 1]

    for i in range(len(a)):

        if a[i] == k:

            return b[i]

    return 'missing'


a = first_val(sheet_overview.iloc[:, [1,3]].values, '1. 2.')


查看完整回答
反对 回复 2021-08-17
  • 1 回答
  • 0 关注
  • 264 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
微信客服

购课补贴
联系客服咨询优惠详情

帮助反馈 APP下载

慕课网APP
您的移动学习伙伴

公众号

扫描二维码
关注慕课网微信公众号