Displaying certain vectors in a 2D Python list

Question

I have the following situation: - I have a variable called (vector_distances). This variable receives a 2D list with several vectors. - I wish it to be displayed, only vectors whose sum of their indices is less than 100. - I have the code below, but you are not doing what I need.

---

def funcao_fitness():
    fitness = 0
    i = 0
    vetor_distancias = [[10,20,30,40],[50,60,70,80],[5,6,7,8],[100,200,300,400],[9,15,25,30]]
    for i in vetor_distancias:
        fitness = (fitness + i) if i <= 100
        print (fitness[i])
    return fitness
print (funcao_fitness())

---

Thanks for your help.

Answer 1

You can do it like this:

def funcao_fitness():
    fitness = []
    vetor_distancias = [[10,20,30,40],[50,60,70,80],[5,6,7,8],[100,200,300,400],[9,15,25,30]]
    for subl in vetor_distancias:
        if sum(subl) <= 100:
            fitness.append(subl)
    return fitness
print funcao_fitness()

Or with list compreension:

def funcao_fitness():
    vetor_distancias = [[10,20,30,40],[50,60,70,80],[5,6,7,8],[100,200,300,400],[9,15,25,30]]
    return [subl for subl in vetor_distancias if sum(subl) <= 100]
print funcao_fitness()

The function will return a list of all sublists whose sum of their values is <= 100

Answer 2

Good as I was trying to solve also for study, I decided to post post...

array = [[10,20,30,40],[50,60,70,80],[5,6,7,8],[100,200,300,400],[9,15,25,30]]
result = [i for i in array if sum(i) < 100]
print(result)

The only difference from ( @Miguel’s reply) will be the same result..

Where in the solution presented the return will be:

[[10, 20, 30, 40], [5, 6, 7, 8], [9, 15, 25, 30]]

And the right thing would be:

[[5, 6, 7, 8], [9, 15, 25, 30]]

But it could be a misinterpretation... due to the operator <= where the correct statement is <...

See on Ideone