numpy where使用方法以及多條件組合使用

# coding:utf-8

import numpy as np

x = np.array([[1, -1], [-1, 1]])
print(x)
"""
[[ 1 -1]
 [-1  1]]
"""

# 將第一列中的-1替換程100
x[:, 0] = np.where(x[:, 0] == -1, 100, x[:, 0])
print(x)
"""
[[  1  -1]
 [100   1]]
"""

# 將第一行中的-1替換成200
x[0, :] = np.where(x[0, :] == -1, 200, x[0, :])
print(x)
"""
[[  1 200]
 [100   1]]
"""

# 將除去100和200之外的所有元素替換成300
x = np.where((x != 100) & (x != 200), 300, x)
print(x)
"""
[[300 200]
 [100 300]]
"""

# 將100和200替換成400
x = np.where((x == 100) | (x == 200), 400, x)
print(x)
"""
[[300 400]
 [400 300]]
"""

參考鏈接