numpy where使用方法以及多条件组合使用

最新推荐文章于 2024-11-10 16:15:53 发布

原创最新推荐文章于 2024-11-10 16:15:53 发布 · 9.2k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy.where #多条件组合

python编程专栏收录该内容

50 篇文章

订阅专栏

本文详细介绍使用Python的NumPy库进行数组元素条件性替换的方法。通过实例展示如何针对特定值进行替换，包括多条件判断，如将指定值替换成其他数值，实现对数组数据的灵活操作。

部署运行你感兴趣的模型镜像

# 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]]
"""

参考链接

您可能感兴趣的与本文相关的镜像