'''
引数に整数 を取り、
iから始まる連番の整数からなる配列をi番目の行として持つ
の2次元配列を返す関数 range_square_matrix() を、
arange() を用いて定義してください。
'''
# 解答
def arange_square_matrix(i):
return np.array([np.arange(k, k + i) for k in range(i)])
# 模範解答
def arange_square_matrix(n):
return np.array([np.arange(i, n+i) for i in range(n)])
