python - 我们如何通过比较列索引 2 的值和字符串中的单词“已批准”来从 numpy ndarray 中选择所有行

Question

['1' 'Approved by person XYZ' 'date'] ['1' 'Approved by person ABC' 'date'] ['1' 'Approved by person PQR' 'date'] ['1' 'Approved by person STD' 'date']

这是我的数组，我只想选择那些包含“已批准”的记录。我知道这可能是一个愚蠢的问题，因为我在 ndarray 中使用字符串值，但仍然想知道。

标签： pythonnumpy-ndarray

Answer 1

arr = [['1', 'Approved by person XYZ', 'date'], ['1', 'Approved by person ABC', 'date'], ['1', 'Approved by person PQR', 'date'], ['1', 'Approved by person STD', 'date']]

[record for record in arr if 'Approved' in record[1]]

适用于 np.arrays 和嵌套列表。返回记录列表