要提取含有特定數字的行,可以使用numpy的條件索引。具體步驟如下:
import numpy as np
arr
,包含多個行和列。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 = 1
result = arr[arr[:, column_index] == 5]
print(result)
輸出結果為:
[[4 5 6]]