Làm thế nào để lọc một mảng numpy bằng cách sử dụng các giá trị của mảng khác?

Question

Tôi có hai mảng NumPy, ví dụ:

a = [1,2,3,4,5] 

và một mảng lọc, ví dụ:

f = [False, True, False, False, True] 

len(a) == len(f) 

Làm thế nào tôi có thể nhận được một mảng NumPy mới chỉ có các giá trị trong một nơi chỉ số tương tự ở f là True? Trong trường hợp của tôi: [2, 5].

Theo giải pháp được chấp nhận (với các giá trị khác nhau):

>>> a = numpy.array([ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) 
>>> b = numpy.array([True, False, True, False, True, False, True, False, True, False]) 
>>> a[b] 
array([1, 3, 5, 7, 9]) 

Answer 1

NumPy hỗ trợ boolean indexing

a[f] 

này giả định rằng a và f là mảng NumPy chứ không phải là danh sách Python (như trong câu hỏi). Bạn có thể chuyển đổi với f = np.array(f).

Answer 2

Nếu bạn chưa cần mảng NumPy, đây là một danh sách đơn giản:

import itertools 
print itertools.compress(a, f) 

Đối với tiền 2,7 phiên bản của python, bạn phải cuộn của riêng bạn (xem thủ công):

def compress(data, selectors): 
    return (d for d, s in itertools.izip(data, selectors) if s)