Python NumPy take方法用法及代碼示例

例子

訪問矩陣中的單個值

x = [[4,5,6], [7,8,9]]
np.take(x, 1)



5

請記住，如果您為第二個參數提供 number 類型，則 Numpy 會將您的輸入數組視為展平數組，即 x=[4,5,6,7,8,9] 。在此示例中，我們采用該展平數組的第一個索引，因此返回值 5。

訪問矩陣的單行

要訪問矩陣的行，請提供參數 axis=0 。

x = [[4,5,6], [7,8,9]]
np.take(x, 1, axis=0)



array([7, 8, 9])

為了澄清這段代碼的作用，我們隻是獲取第二行(即行也遵循索引順序，所以這就是我們將 1 設置為第二個參數的原因)：

訪問矩陣的單列

要訪問矩陣的列，請提供參數 axis=1 。

x = [[4,5,6], [7,8,9]]
np.take(x, 1, axis=1)



array([5, 8])

為了澄清這段代碼的作用，我們隻是獲取第二列(即列也遵循索引順序，所以這就是我們將 1 設置為第二個參數的原因)：

訪問矩陣的多行

要訪問矩陣的多列，請提供參數 axis=0 ，並為第二個參數提供類型 array-like ：

x = [[4,5,6], [7,8,9]]
np.take(x, [0,1], axis=0)



array([[4, 5, 6],
       [7, 8, 9]])

我們正在訪問第一行和第二行，如下所示：

訪問矩陣的多列

要訪問矩陣的多列，請提供參數 axis=1 ，並為第二個參數提供類型 array-like ：

x = [[4,5,6], [7,8,9]]
np.take(x, [0,2], axis=1)



array([[4, 6],
       [7, 9]])

我們正在訪問第一列和第三列，如下所示：