def get_label(data):
    data = data[:, 3:4].flatten().tolist()
    data = [(1 if (int(d) > 0) else 0) for d in data]
    return data
def get_label(data):
    data = data[:, 3:4].flatten().tolist()
    data = [(1 if (int(d) > 0) else 0) for d in data]
    return data