如何让numpy.where()只返回满足条件的元素？

numpy.where()函数是一个用于根据条件返回数组中元素的函数。默认情况下，它返回满足条件的元素的索引。如果你想让numpy.where()只返回满足条件的元素而不是索引，可以通过使用numpy.extract()函数来实现。

numpy.extract()函数接受一个条件和一个数组作为参数，并返回满足条件的元素。下面是一个示例：

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
condition = arr > 2

result = np.extract(condition, arr)
print(result)

输出结果为：

[3 4 5]

在这个例子中，我们首先创建了一个数组arr，然后定义了一个条件condition，即数组中大于2的元素。然后，我们使用np.extract()函数将满足条件的元素提取出来并存储在result变量中。最后，我们打印出result的值，即满足条件的元素。

需要注意的是，np.extract()函数返回的是一个新的数组，而不是原始数组中的视图或副本。因此，对返回的数组进行修改不会影响原始数组。

推荐的腾讯云相关产品：腾讯云云服务器（CVM）和腾讯云对象存储（COS）。

• 腾讯云云服务器（CVM）：腾讯云提供的弹性计算服务，可快速创建和管理云服务器实例，支持多种操作系统和应用场景。详情请参考腾讯云云服务器（CVM）产品介绍。
• 腾讯云对象存储（COS）：腾讯云提供的高可靠、低成本的云端存储服务，适用于存储和处理大规模非结构化数据。详情请参考腾讯云对象存储（COS）产品介绍。