如何获取numpy数组中第一个元素满足要求的特定行或列？

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

condition = arr[:, 0] >= 5  # 获取第一列元素大于等于5的行的条件索引
filtered_rows = arr[condition]  # 获取满足条件的行
first_row = filtered_rows[0]  # 获取第一个满足条件的行

print(first_row)

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

condition = arr[0, :] >= 5  # 获取第一行元素大于等于5的列的条件索引
filtered_columns = arr[:, condition]  # 获取满足条件的列
first_column = filtered_columns[:, 0]  # 获取第一个满足条件的列

print(first_column)