如何在Python中提取3D矩阵的上三角形并放入向量中？

问：

我是一个新的python学习者。我有一个 3d 矩阵，它是 454x454x1172。我想得到它的上三角形，并将其放入一个新的向量中。我该怎么做？

我试过这个。

upper_tsw = tsw[np.triu_indices(3, 1)]
print(upper_tsw)

返回的形状为 （3， 1172）。我认为这是不对的。

另一个我试过了;

shape = dfc_tsw.shape

mask = np.triu(np.ones(shape[1:]), k=1)

result =dfc_tsw[:, mask.astype(bool)]

print(result)
df = pd.DataFrame(result)
df

它的形状是（454,428803）。