python – 使用索引列表从字典中获取值

发布时间：2020-12-20 12:14:31 所属栏目：Python 来源：网络整理

导读：给出一个清单： x = [0.0,0.87,0.0,0.32,0.46,0.10,0.0] 我想得到所有非0值的索引,并将它们存储在d [‘inds’]中然后使用d [‘inds’]中的索引遍历x列表并获取值. 所以我会得到类似的东西： d['inds'] = [1,5,6,9]d['vals'] = [0.87,0.10] 我已经使用了索引

给出一个清单：

x = [0.0,0.87,0.0,0.32,0.46,0.10,0.0]

我想得到所有非0值的索引,并将它们存储在d [‘inds’]中
然后使用d [‘inds’]中的索引遍历x列表并获取值.
所以我会得到类似的东西：

d['inds'] = [1,5,6,9]
d['vals'] = [0.87,0.10]

我已经使用了索引：

d['inds'] = [i for i,m in enumerate(x) if m != 0]

但我不知道如何得到d [‘vals’]

解决方法

d['vals'] = [x[i] for i in d['inds']]

更好的是,同时做两件事：

vals = []
inds = []
for i,v in enumerate(x):
    if v!=0:
        vals.append(v)
        inds.append(i)
d['vals']=vals
d['inds']=inds

要么

import numpy as np
d['inds'],d['vals'] = np.array([(i,v) for i,v in enumerate(x) if v!=0]).T

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!