创建一个分块连接的矩阵，就像楼梯MATLAB一样

import numpy as np

def create_staircase_matrix(n, m):
    result = np.zeros((n, n))  # 创建空的结果矩阵

    for i in range(n // m):
        block = np.zeros((m, m))  # 创建分块矩阵
        block[np.triu_indices(m)] = 1  # 填充楼梯形状

        result[i*m:(i+1)*m, i*m:(i+1)*m] = block  # 将分块矩阵连接到结果矩阵中

    return result

# 示例调用
n = 6  # 楼梯矩阵的大小
m = 2  # 分块的大小
result_matrix = create_staircase_matrix(n, m)
print(result_matrix)

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