返回索引和值作为字典

Question

我正在尝试以字典格式返回非零值及其索引。 例如，如果我有setuptools，则输出将为[0 3 0 6 9]和d['ind'] = [1,3,4] 到目前为止，这是我尝试过的。

d['val'] = [3,6,9]

Answer 1

def inds_vals(x):
    #assert type(x) is list
    d = {'inds': [], 'vals': []}
    for i, v in enumerate(x):
        if v != 0:
            d['inds'].append(i)
            d['vals'].append(v)
    return d

Answer 2

我想这就是你要问的：

def inds_vals(x):
    d = {'inds': [], 'vals': []}
    j=0
    for j,i in enumerate(x):
        if i != 0:
            d['inds']+=[j]
            d['vals']+=[i]
            j = j+1
    return d
print(inds_vals([0,3,0,6,9]))