numpy怎么提取含有特定数字的行

要提取含有特定数字的行，可以使用numpy的条件索引。具体步骤如下：

导入numpy库：import numpy as np创建一个numpy数组，假设名为arr，包含多个行和列。使用条件索引提取含有特定数字的行。假设要提取含有数字5的行，可以使用以下代码：result = arr[arr[:, column_index] == 5]，其中column_index是要检查的列的索引。result即为提取的结果，它是一个包含特定数字的行的numpy数组。

以下是一个完整的示例代码：

import numpy as np# 创建一个示例数组arr = np.array([[1, 2, 3],                [4, 5, 6],                [7, 8, 9]])# 提取含有数字5的行column_index = 1result = arr[arr[:, column_index] == 5]print(result)

输出结果为：

[[4 5 6]]