工作中用到的场景：批量替换和找符合条件的下标

最新推荐文章于 2024-09-15 22:31:42 发布

隔壁老王•天府认证

最新推荐文章于 2024-09-15 22:31:42 发布

阅读量128

点赞数

分类专栏： Numpy 文章标签： python 深度学习概率论

本文链接：https://blog.csdn.net/xiami_tao/article/details/125189761

版权

NumPy 批量操作数据替换坐标查找矩阵

关键词由CSDN通过智能技术生成

Numpy 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

# encoding: utf-8
import numpy as np

# 批量替换 - 类似sql
a = np.array([2, 3, 4, 5, 5, 5, 5, 7])
print(a)
a = np.where(a == 5, 10 * a, a)
print(a)

# 找到满足条件的坐标 -- 很重要的应用
print("*" * 50)

my_matrix = np.array([
    [1, 2, 3, 4],
    [2, 2, 3, 4],
    [3, 2, 3, 4],
])
value2 = np.where(my_matrix == 2)
print(value2)
print("-" * 50)

position_list = []
for i in range(len(value2[0])):
    position_list.append([value2[0][i], value2[1][i]])

print(position_list)


输出：
[2 3 4 5 5 5 5 7]
[ 2  3  4 50 50 50 50  7]
**************************************************
(array([0, 1, 1, 2], dtype=int64), array([1, 0, 1, 1], dtype=int64))
--------------------------------------------------
[[0, 1], [1, 0], [1, 1], [2, 1]]