Double loop with multiple "if" conditionals

Question

I have two arrays with "a" which corresponds to "coordinates" in my problem and b corresponds to specific values of coordinates. I m trying to know the lines of "a" where i got all the 3 values that is in "b" as example i would like to print the line of [2,4,6] as i have them in "b" but nothing appears...there is a mistake...

import numpy as np

a = np.array([[1.,2.,3.],[4.,5.,6.],[2.,4.,6.]])

b = np.array([2,4,6,8,10])

for i in range(0,a.shape[0]):
    for j in range(0,b.shape[0]):
        if (b[j]==a[i,0] and b[j]==a[i,1]):
            print i

Martín Muñoz del Río · Accepted Answer · 2015-11-03 16:16:17Z

2

Try this:

for i in range(0,a.shape[0]):
    if (a[i,0] in b and a[i,1] in b and a[i,2] in b):
        print a[i]

answered Nov 3, 2015 at 16:16

Martín Muñoz del Río

1,96813 silver badges14 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

plonser · Accepted Answer · 2015-11-03 16:20:45Z

1

Since you are already using numpy this here should also work

import numpy as np

a[np.all(np.in1d(a,b).reshape(a.shape),axis=1)]

answered Nov 3, 2015 at 16:20

plonser

3,3632 gold badges21 silver badges23 bronze badges

Collectives™ on Stack Overflow

Double loop with multiple "if" conditionals

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related