获取列包含字符串的numpy数组中的行

import numpy as np

arr = np.array([['apple', 'banana', 'cherry'],
               ['orange', 'banana', 'kiwi'],
               ['apple', 'grape', 'mango']])

target_col = 0  # 假设目标列为第一列

target_str = 'apple'  # 假设要筛选的字符串为'apple'
filtered_rows = arr[arr[:, target_col] == target_str]

print(filtered_rows)

import numpy as np

arr = np.array([['apple', 'banana', 'cherry'],
               ['orange', 'banana', 'kiwi'],
               ['apple', 'grape', 'mango']])

target_col = 0
target_str = 'apple'

filtered_rows = arr[arr[:, target_col] == target_str]
print(filtered_rows)

[['apple' 'banana' 'cherry']
 ['apple' 'grape' 'mango']]