Finding specific array or list in a numpy array

Question

I have a list of numpy arrays, which consists of all possible configurations of 0 and 1 in a 10-pixel arrays. I'm trying to determine the number of arrays that have specific group of 1s for more than two 1s. For example, the array is [1,0,0,1,1,1,1,1,0,1]. I want to determine this array has five 1s as a block. Another example, the array is [1,1,1,0,1,1,1,1,1,1]. I want to find the block as six 1s instead of three 1s block. I couldn't find a way to do this.

Here is the code I generate the list of all possible arrays:

import numpy as np
from itertools import product

all_arrays = np.array(list(product([0,1], repeat=10)))

vencaslac · Accepted Answer · 2022-11-01 13:49:32Z

1

convert the array to a string and check for membership along the lines of:

str_array = ''.join(my_array)
return 5*'1' in str_array

answered Nov 1, 2022 at 13:49

vencaslac

2,8811 gold badge21 silver badges32 bronze badges

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

Comments

Anubhav Sharma · Accepted Answer · 2022-11-01 13:55:59Z

1

Try below approach. Its not very efficient, but it should work.

from functools import reduce
import numpy as np
from itertools import product

n=int(input("How many 1s you want?"))
all_arrays = list(filter(lambda x : n*"1" in x and (n+1)*"1" not in x,map(lambda e : reduce(lambda x,y : str(x)+str(y), e), product([0,1], repeat=10))))
print(len(all_arrays))

answered Nov 1, 2022 at 13:55

Anubhav Sharma

5781 gold badge4 silver badges18 bronze badges

3 Comments

sonido Over a year ago

I was trying to determine this number of 1-blocks for each of the array, instead of generating with a specific number of 1s-block. The solution above worked for this. Thank you for your code though, I can use it for something else!

Anubhav Sharma Over a year ago

So, just to clarify. You want the count but you are trying to avoid generating the arrays, right?

sonido Over a year ago

Yes, that is right. I'm trying to count the number of 1s in a group of 1s for each array in the list

Collectives™ on Stack Overflow

Finding specific array or list in a numpy array

2 Answers 2

Comments

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related