Python Numpy any Filter

Numpy의 any를 이용한 필터

import numpy as np

num_in_array = np.array([0])

x = np.array([
    [2 ,33 ,1 ,6 ,24 ,0],
    [2 ,56 ,2 ,3 ,8 ,5],
    [1 ,2 ,12 ,65 ,4 ,1],
    [9 ,22 ,77 ,4 ,3 ,10],
    [15 ,0 ,2 ,73 ,2 ,3],
    [2 ,42 ,19 ,11 ,55 ,0]
])

# Filter by num_in_array
if num_in_array is not None:
	print(x[:, 5:6] == num_in_array)
    print((x[:, 5:6] == num_in_array).any(1))
	# 마지막 열이 num_in_array안의 숫자와 하나라도 일치할 경우
    # any의 매개변수는 axis
    x = x[(x[:, 5:6] == num_in_array).any(1)]
    
print(x)

'''
결과
[[ True]
 [False]
 [False]
 [False]
 [False]
 [ True]]
[ True False False False False  True]
[[ 2 33  1  6 24  0]
 [ 2 42 19 11 55  0]]
'''

'''
num_in_array = np.array([0, 1])일 경우
[[ True False]
 [False False]
 [False  True]
 [False False]
 [False False]
 [ True False]]
[ True False  True False False  True]
[[ 2 33  1  6 24  0]
 [ 1  2 12 65  4  1]
 [ 2 42 19 11 55  0]]
 '''