np.extract函数

np.extract(y==0, y)就是从一个NumPy数组y中提取值等于0的元素。

假设我们有一个NumPy数组，表示一组学生的分数，我们希望提取分数为0的学生。

import numpy as np

# 假设这是学生的分数数组
scores = np.array([90, 0, 75, 0, 82, 95, 0])

# 使用布尔索引提取分数为0的学生
zero_scores = scores[scores == 0]

print("分数为0的学生：", zero_scores)
# 分数为0的学生: [0 0 0]

在这个例子中，我们首先导入了NumPy库，并创建了一个名为scores的NumPy数组，用于存储学生的分数。接着，我们使用布尔索引scores == 0从数组scores中提取分数等于0的学生，将结果存储在zero_scores变量中。最后，我们打印zero_scores以查看提取到的分数。

上面是从一个NumPy数组y中提取值等于0的元素，为了使用NumPy实现这一点，你可以使用布尔索引。上面的代码首先导入了NumPy库，并创建了一个名为scores的NumPy数组。然后，我们使用布尔索引scores == 0从数组y中提取等于0的元素，将结果存储在zero_scores变量中。