python where
时间: 2023-08-22 18:10:38 浏览: 88
在Python中,`np.where`函数用于从数组中返回满足特定条件的元素的索引位置。它可以有两种不同的用法。一种是`np.where(cond)`,其中`cond`是一个条件表达式,返回满足条件的元素的索引位置。另一种是`np.where(cond, x, y)`,其中`cond`是一个条件表达式,`x`和`y`是两个数组,返回满足条件的元素来自`x`数组的索引位置,不满足条件的元素来自`y`数组的索引位置。\[2\]
举个例子,假设我们有一个数组`arr`,如`arr=np.array(\[1, 5, 7, 4, 3, 1, 9\])`,我们想找到其中等于特定值的元素的索引位置。我们可以使用`np.where(arr==searched_value\[:,None\])\[-1\]`来实现,其中`searched_value`是一个数组,如`searched_value=np.array(\[0, 5, 1\])`。这将返回满足条件的元素在`arr`中的索引位置,即`array(\[1, 0, 5\], dtype=int64)`。\[3\]
#### 引用[.reference_title]
- *1* *2* [一文让你彻底弄懂【Python中的where函数】](https://blog.csdn.net/qq_32532663/article/details/125247125)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [numpy 查找数组中的多个元素索引](https://blog.csdn.net/u014426939/article/details/109737841)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文