Python numpy matrix.squeeze用法及代碼示例

本文簡要介紹 python 語言中 numpy.matrix.squeeze 的用法。

用法:

matrix.squeeze(axis=None)

返回一個可能重新整形的矩陣。

有關更多文檔，請參閱 numpy.squeeze 。

參數：

axis： 無或int 或整數元組，可選

選擇形狀中長度為 1 的軸的子集。如果選擇了形狀條目大於 1 的軸，則會引發錯誤。

返回：

squeezed： 矩陣

矩陣，但如果它的形狀為 (N, 1)，則為 (1, N) 矩陣。

注意：

如果 m 有單列，則該列作為矩陣的單行返回。否則返回 m。返回的矩陣或者是 m 本身，或者是 m 的視圖。提供軸關鍵字參數不會影響返回的矩陣，但可能會引發錯誤。

例子：

>>> c = np.matrix([[1], [2]])
>>> c
matrix([[1],
        [2]])
>>> c.squeeze()
matrix([[1, 2]])
>>> r = c.T
>>> r
matrix([[1, 2]])
>>> r.squeeze()
matrix([[1, 2]])
>>> m = np.matrix([[1, 2], [3, 4]])
>>> m.squeeze()
matrix([[1, 2],
        [3, 4]])