Python arg_aggregate示例

示例#1

def test_arg_reduction():
    pairs = [([4, 3, 5], [10, 11, 12]), ([3, 5, 1], [1, 2, 3])]
    result = arg_aggregate(np.min, np.argmin, (100, 100), pairs)
    assert eq(result, np.array([101, 11, 103]))

示例#2

文件： test_reductions.py 项目： BabeNovelty/dask

def test_arg_reduction():
    pairs = [([4, 3, 5], [10, 11, 12]),
             ([3, 5, 1], [1, 2, 3])]
    result = arg_aggregate(np.min, np.argmin, (100, 100), pairs)
    assert eq(result, np.array([101, 11, 103]))