二维数组问题

huanghaoyang199

目前有随机的正方形矩阵，元素只有数字0和1，若出现由1组成三角形需将其尺寸和数量print。比如：              

       1

1     1    1 尺寸为2（遍历几行出现size就是几）

                                                 1

                                            1    1   1   

                                        1   1   1    1   1尺寸为3

这些大小不同三角形潜藏在0，1随机矩阵中，求各种size的数量，

并print为（size，amount）有多少种size全部列出

例如  1     1      1       1     1

          0     1      1      1    1

           1      1     1     1    1

           1     1    1     1      1

这个矩阵print为[（2,,(3,2)].想了三天了还是想不出来，求各位大佬出手！

huanghaoyang199

会的私信我，红包多多的！

dahe_1984

import numpy as np

a=np.random.randint(0,2,size=[10,10])
print(a)

def oneTriangle(list):
    tri=np.ones(list,dtype=int)
    tri_upper_no_diag=np.triu(tri,k=1)[::-1]
    tri_lower_no_diag=np.tril(tri,k=0)

    a=np.concatenate((tri_upper_no_diag,tri_lower_no_diag),axis=1)

    b=np.delete(a,0,axis=1)

    return b

list=[2,2]
b = oneTriangle(list)

def subsearch(a, b):
    number = 0
    x_b, y_b = b.shape

    for i in range(0,a.shape[0]):
        for j in range(0,a.shape[1]):
            #print(i,j)
            c = a[ii+x_b), jj+y_b)]
            if((c.shape == b.shape) and ((c == b).all())):
                print(c,'\n', b)
                number = number + 1
    return number

number = subsearch(a, b)
print(number)

dahe_1984

[[1 0 0 1 1 1 0 0 0 0]
[0 0 0 1 1 0 1 1 1 0]
[0 1 1 1 0 1 1 1 1 0]
[1 0 1 0 1 0 1 0 0 0]
[0 0 0 0 1 0 0 0 1 0]
[0 1 1 0 1 0 0 1 0 0]
[0 1 0 1 0 1 1 1 0 0]
[1 0 1 1 0 0 1 0 0 0]
[1 0 1 0 1 0 0 0 0 1]
[0 1 0 1 1 1 1 1 1 0]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
1

dahe_1984

[[0 0 1 0 1 0 1 1 1 1 0 0 1 1 1 1 0 1 1 1]
[0 1 0 1 1 1 1 1 1 1 1 0 0 0 1 0 0 1 0 1]
[1 1 0 0 0 1 1 1 1 0 0 0 1 0 0 1 0 1 0 1]
[1 1 1 1 1 0 1 1 1 1 1 0 0 1 1 0 0 0 1 1]
[0 0 0 0 1 1 0 1 0 1 1 1 1 1 0 0 1 1 1 0]
[1 1 0 1 0 0 1 1 1 1 0 1 0 1 0 0 1 1 0 0]
[0 1 0 0 1 0 1 0 1 0 0 0 0 0 0 0 1 1 0 0]
[0 0 1 1 1 1 0 0 1 1 1 0 1 1 1 0 1 0 0 0]
[1 0 1 0 0 1 1 1 0 0 0 1 0 1 1 0 1 1 0 0]
[1 1 0 1 0 1 0 1 0 0 1 0 1 0 0 0 0 1 1 1]
[1 0 0 0 0 0 0 1 0 1 1 1 1 0 1 0 0 1 1 0]
[0 0 1 0 1 1 0 1 1 1 1 1 1 0 1 1 0 1 1 1]
[1 0 1 1 0 0 1 0 1 0 0 1 1 1 0 1 0 1 1 1]
[0 1 0 0 0 0 0 1 1 0 0 0 1 0 0 1 1 0 0 1]
[1 1 0 0 0 1 0 0 1 1 1 0 0 1 0 1 1 1 0 0]
[1 0 1 1 1 1 1 0 1 1 1 0 0 1 1 0 0 1 0 1]
[1 1 1 0 0 0 1 1 1 0 0 0 0 0 0 1 0 1 1 0]
[0 0 0 0 0 1 1 1 0 0 0 0 0 1 0 0 1 0 1 0]
[1 0 0 1 0 0 1 0 0 0 1 0 0 1 1 0 1 1 1 0]
[0 0 0 0 0 0 0 0 0 1 0 1 0 0 1 1 1 1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
[[0 1 0]
[1 1 1]]
5

dahe_1984

具体怎么循环你在自己看吧。

dahe_1984

1. import numpy as np
   
2. import math
   
3. 
   
4. a=np.random.randint(0,2,size=[20,20])
   
5. print(a)
   
6. x_a, y_a = a.shape
   
7. 
   
8. def oneTriangle(list):
   
9.     tri=np.ones(list,dtype=int)
   
10.     tri_upper_no_diag=np.triu(tri,k=1)[::-1]
    
11.     tri_lower_no_diag=np.tril(tri,k=0)
    
12. 
    
13.     a=np.concatenate((tri_upper_no_diag,tri_lower_no_diag),axis=1)
    
14. 
    
15.     b=np.delete(a,0,axis=1)
    
16. 
    
17.     return b
    
18. 
    
19. def subsearch(a, b):
    
20.     number = 0
    
21.     x_b, y_b = b.shape
    
22. 
    
23.     for i in range(0,a.shape[0]):
    
24.         for j in range(0,a.shape[1]):
    
25.             #print(i,j)
    
26.             c = a[i:(i+x_b), j:(j+y_b)]
    
27.             if((c.shape == b.shape) and ((c == b).all())):
    
28.                 #print(c,'\n', b)
    
29.                 number = number + 1
    
30.     return (number, x_b)
    
31. 
    
32. min_value = min(x_a, y_a)
    
33. for i in range(2, min_value):
    
34.     b = oneTriangle([i, i])
    
35.     #print(b)
    
36.     number = subsearch(a, b)
    
37.     print(number)

复制代码