numpy使用where函数定位满足条件的下标

官网DOC:https://numpy.org/doc/stable/reference/generated/numpy.where.html

np.where一般有两个用法:

  1. 当成if else的简写形式
  2. 得到满足条件的下标,方便后续操作

示例

当成if else的简写形式:

import numpy as np

my_array = np.array([1, 2, 3, 4, 5])
value = np.where(my_array > 3, True, False)  # 条件为真就返回第二个参数True,为假就返回第三个参数False
print(value)  # [False False False  True  True]

得到满足条件的下标:

my_matrix = np.array([
    [1, 2, 3, 4],
    [2, 2, 3, 4],
    [3, 2, 3, 4],
])
value2 = np.where(my_matrix > 2)
print(value2)
# (array([0, 0, 1, 1, 2, 2, 2]), array([2, 3, 2, 3, 0, 2, 3]))
# 这里第1个array表示第一个维度的坐标
# 这里第2个array表示第二个维度的坐标
# 所以value2表示下标为[0,2],[0,3](这是my_matrix的第一行1234),[1,2],[1,3](这是my_matrix的第二行2234),[2,0],[2,2],[2,3](这是my_matrix的第三行3234)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

呆萌的代Ma

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值