numpy.flat

>>> x = np.arange(1, 7).reshape(2, 3)
>>> x
array([[1, 2, 3],
       [4, 5, 6]])
>>> x.flat[3] # 返回重组后的一维数组下标为3的元素
4
>>> x.T
array([[1, 4],
       [2, 5],
       [3, 6]])
>>> x.T.flat[3] # 返回x的转置重组后的一维数组下标为3的元素
5
>>> x.flat = 3 # 将数组的元素均变为3
>>> x
array([[3, 3, 3],
       [3, 3, 3]])
>>> x.flat[[1,4]] = 1 # 将数组重组后的一维数组小标为1,4的元素变为1
>>> x
array([[3, 1, 3],
       [3, 1, 3]])