Filtering a list based on a list of booleans

后端 未结 6 1242
清歌不尽
清歌不尽 2020-11-27 10:14

I have a list of values which I need to filter given the values in a list of booleans:

list_a = [1, 2, 4, 6]
filter = [True, False, True, False]
6条回答
  •  时光说笑
    2020-11-27 11:14

    To do this using numpy, ie, if you have an array, a, instead of list_a:

    a = np.array([1, 2, 4, 6])
    my_filter = np.array([True, False, True, False], dtype=bool)
    a[my_filter]
    > array([1, 4])
    

提交回复
热议问题