Numpy -计算矩阵中心不间断行的长度

import numpy as np

# 创建一个示例矩阵
matrix = np.array([[0, 0, 0, 0],
                   [1, 1, 1, 1],
                   [0, 0, 0, 0],
                   [1, 1, 1, 1],
                   [1, 1, 1, 1],
                   [0, 0, 0, 0]])

# 计算矩阵中心不间断行的长度
center_row = matrix.shape[0] // 2  # 计算矩阵中心行的索引
nonzero_rows = np.flatnonzero(matrix.sum(axis=1))  # 找到非零行的索引
continuous_rows = np.split(nonzero_rows, np.where(np.diff(nonzero_rows) != 1)[0] + 1)  # 将连续的行分组

center_continuous_rows = [row for row in continuous_rows if center_row in row]  # 找到包含中心行的连续行组
center_continuous_rows_lengths = [len(row) for row in center_continuous_rows]  # 计算连续行组的长度

print("矩阵中心不间断行的长度：", center_continuous_rows_lengths)