首页 > 解决方案 > 如何为 dask apply_over_axes 调用设置元

问题描述

以下代码

a = da.random.normal(0, 1., size=(100, 100), chunks=(15, 15))
def cut(x,y):
    try:
        res = np.int8(np.digitize(x[0]+x[1], [.25,.50,.75]))
    except:
        res = np.int8(0)
    return np.int8(res)

res = da.apply_over_axes( cut, a, [1])
da.to_hdf5( '/tmp/cat.h5', {'/cat': res})

导致使用 aguments 对 cut 进行初始调用x=[1.], y=0.,这会触发 cut 定义中的 except 语句。

有没有办法直接设置元?

标签: pythondask

解决方案


从 0.18.2 版开始,它看起来不像,这是硬连线的。我建议提出一个问题。


推荐阅读