python numpy庫中where用法

python整段注釋ctrl+/

import numpy as np
np_where = np.where([[false,true],[true,true]],[[1,3],[2,4]],[[4,2],[6,5]])
print(np_where) #[[4 3][2 4]]

np.where(x) x為陣列，表示返回其中非0元素的下標

x = np.array([1,2,3]) #(array([0, 1, 2], dtype=int64),)
y = np.array([[0,0,3],[4,5,0]]) #(array([0, 1, 1], dtype=int64), array([2, 0, 1], dtype=int64))
print(np.where(x))
print(np.where(y))

給where()函式傳入乙個判斷條件，返回true的位置，若見該位置再傳入陣列中則可返回符合條件的元素

x = np.array([1,2,3]) 
y = np.array([[0,0,3],[4,5,]])
print(np.where(x%2 == 1)) #(array([0, 2], dtype=int64),)
print(x[np.where(x%2 == 1)]) #[1
3]print(np.where(y>1)) #(array([0, 1, 1], dtype=int64), array([2, 0, 1], dtype=int64))
print(y[np.where(y>1)]) #[3
45]

where(condition, x=none, y=none)對於條件陣列中等價於true的位置，從第乙個值或陣列中取值進行替換，否則從第二個值或陣列中取值進行替換。

x = np.array([1,2,3])
y = np.array([[0,0,3],[4,5,0]])
print(np.where(x%2==1,x,-x)) #[ 1 -2
3]print(np.where(y, [10,20,30], [40,50,60])) #[[40 50 30] [10 20 60]]

python numpy中nonzero 的用法

當使用布林陣列直接作為下標物件或者元組下標物件中有布林陣列時，都相當於用nonzero 將布林陣列轉換成一組整數陣列，然後使用整數陣列進行下標運算。nonzeros a 返回陣列a中值不為零的元素的下標，它的返回值是乙個長度為a.ndim 陣列a的軸數的元組，元組的每個元素都是乙個整數陣列，其值為...

python numpy庫中where用法

python numpy中nonzero 的用法

python numpy中nonzero 的用法

python numpy中nonzero 的用法

相關推薦