# 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]]
"""
numpy where使用方法以及多条件组合使用
最新推荐文章于 2024-07-30 11:50:19 发布