将索引和值作为字典返回

Question

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

def inds_vals(x):
    
    d = {'inds': [], 'vals': []}
    j=0
    for i in x:
        if i != 0:
            d.update({'inds': j, 'vals':i})
            j = j+1
    return

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]))