Python Numpy numpy.choose()用法及代碼示例

借助Numpy numpy.choose()方法，我們可以通過將包含要選擇的行號索引的參數作為數組傳遞，從多維數組中選擇元素。具有與參數中傳遞的大小相同的輸出數組。

用法： numpy.choose()

返回： 返回元素選擇的數組

範例1：
在此示例中，我們可以借助numpy.choose()方法，我們能夠從多維數組中提取選擇數組。

# import the important module in python 
import numpy as np 
             
# make a matrix with numpy 
gfg = [[1, 2, 3, 4], [3, 1, 5, 6]] 
             
# applying numpy.choose() method 
geeks = np.choose([1, 0, 1, 0], gfg) 
       
print(geeks)

輸出：

array([3 2 5 4])

範例2：

# import the important module in python 
import numpy as np 
             
# make a matrix with numpy 
gfg = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 
             
# applying numpy.choose() method 
geeks = np.choose([2, 0, 1], gfg) 
       
print(geeks)

輸出：

array([7 2 6])

相關用法

注：本文由純淨天空篩選整理自Jitender_1998大神的英文原創作品 Python | Numpy numpy.choose()。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。