numpy 연산

import numpy as np

def solution(info):
    answer = []
    size = np.shape(info)
    for height in range(size[1]):
        if info[0][height]>195.0 or info[0][height]<150.0:
            answer.append(height)
    for weight in range(size[1]):
        if info[1][weight]>=140.0:
            answer.append(weight)
    answer = list(set(answer))
    return answer