Хочешь этого? (уточняет комментарий Кевина)
import numpy as np
A = np.array([[i for i in range(1,198)] for j in range(1,198)])
A
A.shape
будет производить
array([[ 1, 2, 3, ..., 195, 196, 197],
[ 1, 2, 3, ..., 195, 196, 197],
[ 1, 2, 3, ..., 195, 196, 197],
...,
[ 1, 2, 3, ..., 195, 196, 197],
[ 1, 2, 3, ..., 195, 196, 197],
[ 1, 2, 3, ..., 195, 196, 197]])
(197, 197)