check condition for a list of lists

rizwan
11 ●1

I have a list of lists L = [L1, L2, L3,... Lm] where each Li is a list again. I want to check each Li for a condition.

I need only those Li where all elements of Li satisfy a certain property.

I am trying the following but it's not working the way I want.

for i in range(0, len(L)):
    U = L[i]
    for s in U:
          check(s)

But this does not distinguish between the elements of Li.

A little help is needed.

Thanks

1 Answer

Sort by » oldest newest most voted

achrzesz
2803 ●5 ●25 ●61

Learn about the Python command all using help(all).

Below are a couple of examples.

First define a list of lists (of integers):

L1 = [1, 2]
L2 = [-1, 1]
L3 = [3, 1]
L = [L1, L2, L3]

Then keep only those consisting only of positive integers:

using map:

[Li for Li in L if all(map(lambda v: v > 0, Li))]

using list comprehension:

[Li for Li in L if all(v > 0 for v in Li)]

For your use case, this might become

LL = [Li for Li in L if  all(check(s) for s in Li)]

Please start posting anonymously - your entry will be published after you log in or create a new account.

1 follower

Asked: 2022-11-19 23:21:08 +0100

Seen: 275 times

Last updated: Nov 26 '22