Write index from a list in another list

Question

Write index from a list in another list

Navigation

#1 by (4 votes)

2

T = [0,0,0,1,0,0,0,-1,0]
def acoes(T):
    listaAction=[]
    for i in T:
            if(T[i]==0):
                    listaAction.append(T.index(i))

    return listaAction

print(acoes(T))

How do I write list indexes that have value 0 in listAction?

python lista

asked by anonymous 19.10.2017 / 11:53

1 answer

Show Calculator on Stack How to compare vectors in C

score 4 · Accepted Answer

Taking advantage of what you can do this:

def acoes(T):
    listaAction=[]
    for idx, i in enumerate(T):
        if(i==0):
            listaAction.append(idx)
    return listaAction

T = [0,0,0,1,0,0,0,-1,0]
print(acoes(T)) # [0, 1, 2, 4, 5, 6, 8]

acoes(T) will return a list with indices in T whose value is zero

You can do it in a way even more pythonica :

def acoes(T):
    return [idx for idx, i in enumerate(T) if i == 0]

T = [0,0,0,1,0,0,0,-1,0]
print(acoes(T)) # [0, 1, 2, 4, 5, 6, 8]

DEMONSTRATION of both examples