分割数组

def split(arr, cond):
	return [arr[cond], arr[~cond]]

data_true, data_false = split(data_x, data_y == 1)
posted @ 2022-02-09 17:58  小康要好好学习  阅读(36)  评论(0编辑  收藏  举报