I have two arrays, id, and x where id is a unique identifier that tells us that the values in x belong to a specific group. What I want to do is go through the values in x to see if some condition is meet and if so print the corresponding x value. For example
id = np.array([1,1,1,2,2,2,3,3,3,4,4,4,5,5,5])
x = np.array([10,9,6,9,7,1,12,5,10,9,8,4,6,2,1])
counter = 1
for i in range(len(id)):
if id[i] == counter:
for j in range(i,len(id)):
if x[j] > 7:
continue
else:
print(id[i],x[j])
counter += 1
break
prints
1 6
2 7
3 5
4 4
5 6
Now if we instead have
id = np.array([1,1,1,2,2,2,3,3,3,4,4,4,5,5,5])
x = np.array([10,9,6,9,7,1,12,11,10,9,8,4,6,2,1])
The output is
1 6
2 7
3 4
4 4
5 6
Which is not the output I want because 4 is not in the group that has an id value of 3. So my question is how does one only have the condition if x[j] > 7: evaluated if the x values correspond to an id vale that represents it and not skip over that group?
Aucun commentaire:
Enregistrer un commentaire