问Python:计数numpy数组中可变大小的子数组实例

EN

import numpy as np
import itertools
import time

def itertools_solution(full_array):
  trueFalse = full_array == 0
  count = [ sum( 1 for _ in group ) for key, group in itertools.groupby( trueFalse ) if key ]
  above = [val for val in count if val >= 3]
  return len(above)

def looping_solution(full_array):
  total_count = 0
  running_count = 0
  for counter, val in enumerate(full_array):
    if val == 0:
      running_count += 1
      if running_count == 3:
        total_count += 1
    else:
      running_count = 0
  return total_count

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

time_start = time.time()
for cnt in range(1000):
  itertools_solution(a)
print('itertools took %f seconds' % (time.time() - time_start))
time_start = time.time()
for cnt in range(1000):
  looping_solution(a)
print('looping took %f seconds' % (time.time() - time_start))