与 numpy 函数的兼容性

与 numpy 函数的兼容性¶

下表描述了 numpy 和 dask.array 函数之间的兼容性。请注意，两个版本之间可能存在一些不一致。

此表格是手动编译的，可能无法反映当前的 Dask 状态。欢迎贡献更新。

空白条目表示该函数未在 Dask 中实现。
直接实现是直接调用 numpy 函数。
逐元素实现派生自 numpy，但逐元素应用：参数应为 Dask 数组。
Dask 等效是 Dask 实现，相对于 numpy 函数可能会缺少或添加参数。

在 Array API 框架下更深入的比较可参见 Array API 比较仓库。

NumPy	Dask	实现
`numpy.absolute`	`dask.array.absolute` 或 `dask.array.abs`	直接 (ufunc)
`numpy.add`	`dask.array.add`	直接 (ufunc)
`numpy.all`	`dask.array.all` 21	Dask 等效
`numpy.allclose`	`dask.array.allclose`	Dask 等效
`numpy.amax`	`dask.array.max` 21 22	Dask 等效
`numpy.amin`	`dask.array.min` 21 22	Dask 等效
`numpy.angle`	`dask.array.angle` 23	Dask 等效
`numpy.any`	`dask.array.any` 21	Dask 等效
`numpy.append`	`dask.array.append`	Dask 等效
`numpy.apply_along_axis`	`dask.array.apply_along_axis`	Dask 等效
`numpy.apply_over_axes`	`dask.array.apply_over_axes`	Dask 等效
`numpy.arange`	`dask.array.arange`	Dask 等效
`numpy.arccos`	`dask.array.arccos`	直接 (ufunc)
`numpy.arccosh`	`dask.array.arccosh`	直接 (ufunc)
`numpy.arcsin`	`dask.array.arcsin`	直接 (ufunc)
`numpy.arcsinh`	`dask.array.arcsinh`	直接 (ufunc)
`numpy.arctan`	`dask.array.arctan`	直接 (ufunc)
`numpy.arctan2`	`dask.array.arctan2`	直接 (ufunc)
`numpy.arctanh`	`dask.array.arctanh`	直接 (ufunc)
`numpy.argmax`	`dask.array.argmax`	Dask 等效
`numpy.argmin`	`dask.array.argmin`	Dask 等效
`numpy.argpartition`	-
`numpy.argsort`	`dask.array.argtopk` 25
`numpy.argwhere`	`dask.array.argwhere`	Dask 等效
`numpy.around`	`dask.array.around` 23 26 或 `dask.array.round`	Dask 等效
`numpy.array`	`dask.array.array`	Dask 等效
`numpy.array2string`	-
`numpy.array_equal`	-
`numpy.array_equiv`	-
`numpy.array_repr`	-
`numpy.array_split`	-
`numpy.array_str`	-
`numpy.asanyarray`	`dask.array.asanyarray`	Dask 等效
`numpy.asarray`	`dask.array.asarray`	Dask 等效
`numpy.asarray_chkfinite`	-
`numpy.ascontiguousarray`	-
`numpy.asfarray`	-
`numpy.asfortranarray`	-
`numpy.asmatrix`	- 27
`numpy.atleast_1d`	`dask.array.atleast_1d`	Dask 等效
`numpy.atleast_2d`	`dask.array.atleast_2d`	Dask 等效
`numpy.atleast_3d`	`dask.array.atleast_3d`	Dask 等效
`numpy.average`	`dask.array.average`	Dask 等效
`numpy.bartlett`	-
`numpy.bincount`	`dask.array.bincount`	Dask 等效
`numpy.bitwise_and`	`dask.array.bitwise_and`	直接 (ufunc)
`numpy.bitwise_or`	`dask.array.bitwise_or`	直接 (ufunc)
`numpy.bitwise_xor`	`dask.array.bitwise_xor`	直接 (ufunc)
`numpy.blackman`	-
`numpy.block`	`dask.array.block`	Dask 等效
`numpy.bmat`	- 27
`numpy.broadcast`	-
`numpy.broadcast_arrays`	`dask.array.broadcast_arrays`	Dask 等效
`numpy.broadcast_shapes`	-
`numpy.broadcast_to`	`dask.array.broadcast_to`	Dask 等效
`numpy.byte_bounds`	-
`numpy.c_`	-
`numpy.can_cast`	-
`numpy.cbrt`	`dask.array.cbrt`	直接 (ufunc)
`numpy.ceil`	`dask.array.ceil`	直接 (ufunc)
`numpy.choose`	`dask.array.choose` 28	Dask 等效
`numpy.clip`	`dask.array.clip` 23 26	直接 (非 ufunc 逐元素)
`numpy.column_stack`	-
`numpy.common_type`	-
`numpy.compress`	`dask.array.compress` 26	Dask 等效
`numpy.concatenate`	`dask.array.concatenate`	Dask 等效
`numpy.conj`	`dask.array.conj`	直接 (ufunc)
`numpy.conjugate`	`dask.array.conj`	直接 (ufunc)
`numpy.convolve`	-
`numpy.copy`	-
`numpy.copysign`	`dask.array.copysign`	直接 (ufunc)
`numpy.copyto`	-
`numpy.corrcoef`	`dask.array.corrcoef`	Dask 等效
`numpy.correlate`	-
`numpy.cos`	`dask.array.cos`	直接 (ufunc)
`numpy.cosh`	`dask.array.cosh`	直接 (ufunc)
`numpy.count_nonzero`	`dask.array.count_nonzero` 29	Dask 等效
`numpy.cov`	`dask.array.cov` 30	Dask 等效
`numpy.cross`	-
`numpy.cumprod`	`dask.array.cumprod` 23 38	Dask 等效
`numpy.cumsum`	`dask.array.cumsum` 23 38	Dask 等效
`numpy.datetime_as_string`	-
`numpy.deg2rad`	`dask.array.deg2rad`	直接 (ufunc)
`numpy.degrees`	`dask.array.degrees`	直接 (ufunc)
`numpy.delete`	`dask.array.delete`	Dask 等效
`numpy.diag`	`dask.array.diag`	Dask 等效
`numpy.diag_indices`	-
`numpy.diag_indices_from`	-
`numpy.diagflat`	-
`numpy.diagonal`	`dask.array.diagonal`	Dask 等效
`numpy.diff`	`dask.array.diff`	Dask 等效
`numpy.digitize`	`dask.array.digitize` 23	Dask 等效
`numpy.divide`	`dask.array.divide`	直接 (ufunc)
`numpy.divmod`	`dask.array.divmod`	Dask 等效
`numpy.dot`	`dask.array.dot` 26	Dask 等效
`numpy.dsplit`	-
`numpy.dstack`	`dask.array.dstack`	Dask 等效
`numpy.ediff1d`	`dask.array.ediff1d`	Dask 等效
`numpy.einsum`	`dask.array.einsum` 26	Dask 等效
`numpy.einsum_path`	-
`numpy.empty`	`dask.array.empty`	Dask 等效
`numpy.empty_like`	`dask.array.empty_like`	Dask 等效
`numpy.equal`	`dask.array.equal`	直接 (ufunc)
`numpy.exp`	`dask.array.exp`	直接 (ufunc)
`numpy.exp2`	`dask.array.exp2`	直接 (ufunc)
`numpy.expand_dims`	`dask.array.expand_dims`	Dask 等效
`numpy.expm1`	`dask.array.expm1`	直接 (ufunc)
`numpy.extract`	`dask.array.extract`	Dask 等效
`numpy.eye`	`dask.array.eye`	Dask 等效
`numpy.fabs`	`dask.array.fabs`	直接 (ufunc)
`numpy.fill_diagonal`	-
`numpy.fix`	`dask.array.fix`	直接 (非 ufunc 逐元素)
`numpy.flatnonzero`	`dask.array.flatnonzero`	Dask 等效
`numpy.flip`	`dask.array.flip`	Dask 等效
`numpy.fliplr`	`dask.array.fliplr`	Dask 等效
`numpy.flipud`	`dask.array.flipud`	Dask 等效
`numpy.float_power`	`dask.array.float_power`	直接 (ufunc)
`numpy.floor`	`dask.array.floor`	直接 (ufunc)
`numpy.floor_divide`	`dask.array.floor_divide`	直接 (ufunc)
`numpy.fmax`	`dask.array.fmax`	直接 (ufunc)
`numpy.fmin`	`dask.array.fmin`	直接 (ufunc)
`numpy.fmod`	`dask.array.fmod`	直接 (ufunc)
`numpy.frexp`	`dask.array.frexp`	Dask 等效
`numpy.from_dlpack`	-
`numpy.frombuffer`	-
`numpy.fromfile`	-
`numpy.fromfunction`	`dask.array.fromfunction` 31	Dask 等效
`numpy.fromiter`	-
`numpy.frompyfunc`	`dask.array.frompyfunc` 32	Dask 等效
`numpy.fromregex`	-
`numpy.fromstring`	-
`numpy.full`	`dask.array.full`	Dask 等效
`numpy.full_like`	`dask.array.full_like`	Dask 等效
`numpy.gcd`	-
`numpy.genfromtxt`	-
`numpy.geomspace`	-
`numpy.gradient`	`dask.array.gradient` 33	Dask 等效
`numpy.greater`	`dask.array.greater`	直接 (ufunc)
`numpy.greater_equal`	`dask.array.greater_equal`	直接 (ufunc)
`numpy.hamming`	-
`numpy.hanning`	-
`numpy.heaviside`	-
`numpy.histogram`	`dask.array.histogram`	Dask 等效
`numpy.histogram2d`	`dask.array.histogram2d`	Dask 等效
`numpy.histogram_bin_edges`	-
`numpy.histogramdd`	`dask.array.histogramdd` 34	Dask 等效
`numpy.hsplit`	-
`numpy.hstack`	`dask.array.hstack`	Dask 等效
`numpy.hypot`	`dask.array.hypot`	直接 (ufunc)
`numpy.i0`	`dask.array.i0`	直接 (非 ufunc 逐元素)
`numpy.identity`	-
`numpy.imag`	`dask.array.imag`	直接 (非 ufunc 逐元素)
`numpy.in1d`	-
`numpy.indices`	`dask.array.indices`	Dask 等效
`numpy.inner`	-
`numpy.insert`	`dask.array.insert` 35	Dask 等效
`numpy.interp`	-
`numpy.intersect1d`	-
`numpy.invert`	`dask.array.invert` 或 `dask.array.bitwise_not`	直接 (ufunc)
`numpy.is_busday`	-
`numpy.isclose`	`dask.array.isclose`	Dask 等效
`numpy.iscomplex`	`dask.array.iscomplex`	直接 (非 ufunc 逐元素)
`numpy.iscomplexobj`	-
`numpy.isfinite`	`dask.array.isfinite`	直接 (ufunc)
`numpy.isfortran`	-
`numpy.isin`	`dask.array.isin`	Dask 等效
`numpy.isinf`	`dask.array.isinf`	直接 (ufunc)
`numpy.isnan`	`dask.array.isnan`	直接 (ufunc)
`numpy.isnat`	-
`numpy.isneginf`	`dask.array.isneginf`	直接 (ufunc)
`numpy.isposinf`	`dask.array.isposinf`	直接 (ufunc)
`numpy.isreal`	`dask.array.isreal`	直接 (非 ufunc 逐元素)
`numpy.ix_`	-
`numpy.kaiser`	-
`numpy.kron`	-
`numpy.lcm`	-
`numpy.ldexp`	`dask.array.ldexp`	直接 (ufunc)
`numpy.left_shift`	`dask.array.left_shift`	直接 (ufunc)
`numpy.less`	`dask.array.less`	直接 (ufunc)
`numpy.less_equal`	`dask.array.less_equal`	直接 (ufunc)
`numpy.lexsort`	-
`numpy.linspace`	`dask.array.linspace`	Dask 等效
`numpy.load`	-
`numpy.loadtxt`	-
`numpy.log`	`dask.array.log`	直接 (ufunc)
`numpy.log10`	`dask.array.log10`	直接 (ufunc)
`numpy.log1p`	`dask.array.log1p`	直接 (ufunc)
`numpy.log2`	`dask.array.log2`	直接 (ufunc)
`numpy.logaddexp`	`dask.array.logaddexp`	直接 (ufunc)
`numpy.logaddexp2`	`dask.array.logaddexp2`	直接 (ufunc)
`numpy.logical_and`	`dask.array.logical_and`	直接 (ufunc)
`numpy.logical_not`	`dask.array.logical_not`	直接 (ufunc)
`numpy.logical_or`	`dask.array.logical_or`	直接 (ufunc)
`numpy.logical_xor`	`dask.array.logical_xor`	直接 (ufunc)
`numpy.logspace`	-
`numpy.mask_indices`	-
`numpy.mat`	- 27
`numpy.matmul`	`dask.array.matmul`	Dask 等效
`numpy.matrix`	- 27
`numpy.maximum`	`dask.array.maximum`	直接 (ufunc)
`numpy.may_share_memory`	-
`numpy.mean`	`dask.array.mean` 21	Dask 等效
`numpy.median`	`dask.array.median` 36	Dask 等效
`numpy.memmap`	-
`numpy.meshgrid`	`dask.array.meshgrid` 37	Dask 等效
`numpy.mgrid`	-
`numpy.minimum`	`dask.array.minimum`	直接 (ufunc)
`numpy.mintypecode`	-
`numpy.mod`	`dask.array.mod`	直接 (ufunc)
`numpy.modf`	`dask.array.modf`	Dask 等效
`numpy.moveaxis`	`dask.array.moveaxis`	Dask 等效
`numpy.multiply`	`dask.array.multiply`	直接 (ufunc)
`numpy.nan_to_num`	`dask.array.nan_to_num`	直接 (非 ufunc 逐元素)
`numpy.nanargmax`	`dask.array.nanargmax`	Dask 等效
`numpy.nanargmin`	`dask.array.nanargmin`	Dask 等效
`numpy.nancumprod`	`dask.array.nancumprod` 23 38	Dask 等效
`numpy.nancumsum`	`dask.array.nancumsum` 23 38	Dask 等效
`numpy.nanmax`	`dask.array.nanmax` 21 22	Dask 等效
`numpy.nanmean`	`dask.array.nanmean` 21	Dask 等效
`numpy.nanmedian`	`dask.array.nanmedian` 36	Dask 等效
`numpy.nanmin`	`dask.array.nanmin` 21 22	Dask 等效
`numpy.nanpercentile`	`dask.array.nanpercentile`
`numpy.nanprod`	`dask.array.nanprod` 21 22	Dask 等效
`numpy.nanquantile`	`dask.array.nanquantile`
`numpy.nanstd`	`dask.array.nanstd` 21	Dask 等效
`numpy.nansum`	`dask.array.nansum` 21 22	Dask 等效
`numpy.nanvar`	`dask.array.nanvar` 21	Dask 等效
`numpy.ndenumerate`	-
`numpy.ndindex`	-
`numpy.nditer`	-
`numpy.negative`	`dask.array.negative`	直接 (ufunc)
`numpy.nested_iters`	-
`numpy.nextafter`	`dask.array.nextafter`	直接 (ufunc)
`numpy.nonzero`	`dask.array.nonzero`	Dask 等效
`numpy.not_equal`	`dask.array.not_equal`	直接 (ufunc)
`numpy.ogrid`	-
`numpy.ones`	`dask.array.ones`	Dask 等效
`numpy.ones_like`	`dask.array.ones_like`	Dask 等效
`numpy.outer`	`dask.array.outer`	Dask 等效
`numpy.packbits`	-
`numpy.pad`	`dask.array.pad`	Dask 等效
`numpy.partition`	-
`numpy.percentile`	`dask.array.percentile`	Dask 等效
`numpy.piecewise`	`dask.array.piecewise`	Dask 等效
`numpy.place`	-
`numpy.poly`	-
`numpy.poly1d`	-
`numpy.polyadd`	-
`numpy.polyder`	-
`numpy.polydiv`	-
`numpy.polyfit`	-
`numpy.polyint`	-
`numpy.polymul`	-
`numpy.polysub`	-
`numpy.polyval`	-
`numpy.positive`	`dask.array.positive`	直接 (ufunc)
`numpy.power`	`dask.array.power`	直接 (ufunc)
`numpy.prod`	`dask.array.prod`	Dask 等效
`numpy.ptp`	`dask.array.ptp`	Dask 等效
`numpy.put`	-
`numpy.put_along_axis`	-
`numpy.putmask`	-
`numpy.quantile`	`dask.array.quantile`
`numpy.r_`	-
`numpy.rad2deg`	`dask.array.rad2deg`	直接 (ufunc)
`numpy.radians`	`dask.array.radians`	直接 (ufunc)
`numpy.ravel`	`dask.array.ravel` 23 24	Dask 等效
`numpy.ravel_multi_index`	`dask.array.ravel_multi_index`	Dask 等效
`numpy.real`	`dask.array.real`	直接 (非 ufunc 逐元素)
`numpy.real_if_close`	-
`numpy.reciprocal`	`dask.array.reciprocal`	直接 (ufunc)
`numpy.remainder`	`dask.array.remainder`	直接 (ufunc)
`numpy.repeat`	`dask.array.repeat`	Dask 等效
`numpy.require`	-
`numpy.reshape`	`dask.array.reshape`	Dask 等效
`numpy.resize`	-
`numpy.result_type`	`dask.array.result_type`	Dask 等效
`numpy.right_shift`	`dask.array.right_shift`	直接 (ufunc)
`numpy.rint`	`dask.array.rint`	直接 (ufunc)
`numpy.roll`	`dask.array.roll`	Dask 等效
`numpy.rollaxis`	`dask.array.rollaxis`	Dask 等效
`numpy.roots`	-
`numpy.rot90`	`dask.array.rot90`	Dask 等效
`numpy.row_stack`	-
`numpy.save`	-
`numpy.savetxt`	-
`numpy.savez`	-
`numpy.savez_compressed`	-
`numpy.searchsorted`	`dask.array.searchsorted`	Dask 等效
`numpy.select`	`dask.array.select`	Dask 等效
`numpy.setdiff1d`	-
`numpy.setxor1d`	-
`numpy.shape`	`dask.array.shape` 23	Dask 等效
`numpy.shares_memory`	-
`numpy.sign`	`dask.array.sign`	直接 (ufunc)
`numpy.signbit`	`dask.array.signbit`	直接 (ufunc)
`numpy.sin`	`dask.array.sin`	直接 (ufunc)
`numpy.sinc`	`dask.array.sinc`	直接 (非 ufunc 逐元素)
`numpy.sinh`	`dask.array.sinh`	直接 (ufunc)
`numpy.sort`	`dask.array.topk` 25
`numpy.sort_complex`	-
`numpy.source`	-
`numpy.spacing`	`dask.array.spacing`	直接 (ufunc)
`numpy.split`	-
`numpy.sqrt`	`dask.array.sqrt`	直接 (ufunc)
`numpy.square`	`dask.array.square`	直接 (ufunc)
`numpy.squeeze`	`dask.array.squeeze`	Dask 等效
`numpy.stack`	`dask.array.stack`	Dask 等效
`numpy.std`	`dask.array.std` 21	Dask 等效
`numpy.subtract`	`dask.array.subtract`	直接 (ufunc)
`numpy.sum`	`dask.array.sum` 21 22	Dask 等效
`numpy.swapaxes`	`dask.array.swapaxes`	Dask 等效
`numpy.take`	`dask.array.take` 28	Dask 等效
`numpy.take_along_axis`	-
`numpy.tan`	`dask.array.tan`	直接 (ufunc)
`numpy.tanh`	`dask.array.tanh`	直接 (ufunc)
`numpy.tensordot`	`dask.array.tensordot`	Dask 等效
`numpy.tile`	`dask.array.tile`	Dask 等效
`numpy.trace`	`dask.array.trace` 26	Dask 等效
`numpy.transpose`	`dask.array.transpose`	Dask 等效
`numpy.trapz`	-
`numpy.tri`	`dask.array.tri`	Dask 等效
`numpy.tril`	`dask.array.tril`	Dask 等效
`numpy.tril_indices`	`dask.array.tril_indices`	Dask 等效
`numpy.tril_indices_from`	`dask.array.tril_indices_from`	Dask 等效
`numpy.trim_zeros`	-
`numpy.triu`	`dask.array.triu`	Dask 等效
`numpy.triu_indices`	`dask.array.triu_indices`	Dask 等效
`numpy.triu_indices_from`	`dask.array.triu_indices_from`	Dask 等效
`numpy.true_divide`	`dask.array.true_divide`	直接 (ufunc)
`numpy.trunc`	`dask.array.trunc`	直接 (ufunc)
`numpy.union1d`	`dask.array.union1d`	Dask 等效
`numpy.unique`	`dask.array.unique` 39	Dask 等效
`numpy.unpackbits`	-
`numpy.unravel_index`	`dask.array.unravel_index`	Dask 等效
`numpy.unwrap`	-
`numpy.vander`	-
`numpy.var`	`dask.array.var` 21	Dask 等效
`numpy.vdot`	`dask.array.vdot`	Dask 等效
`numpy.vsplit`	-
`numpy.vstack`	`dask.array.vstack` 40	Dask 等效
`numpy.where`	`dask.array.where`	Dask 等效
`numpy.zeros`	`dask.array.zeros`	Dask 等效
`numpy.zeros_like`	`dask.array.zeros_like`	Dask 等效

注脚

21(1,2,3,4,5,6,7,8,9,10,11,12,13,14,15): 不支持 where 参数。
22(1,2,3,4,5,6,7): 不支持 initial 参数。
23(1,2,3,4,5,6,7,8,9,10): 输入必须是 Dask 数组。
24: 不支持 order 参数。
25(1,2): 并行排序操作非常困难。并行友好的替代方法是对 k 个最大元素进行排序。
26(1,2,3,4,5,6): 不支持 out 参数。
27(1,2,3,4): NumPy 中不推荐使用 numpy.matrix，因此无需添加它。
28(1,2): 不支持 mode 参数。
29: 不支持 keepdims 参数。
30: 不支持 fweights, aweights, dtype 参数。
31: 不支持 like 参数。不支持可调用函数。
32: 未实现多于一个输出。
33: 不支持 edge_order 参数。
34: 输入数据（样本）的分块仅沿第 0（行）轴允许。
35: 仅对单调的 obj 参数实现。
36(1,2): 不支持 overwrite_input 参数。
37: 不支持 copy 参数。
38(1,2,3,4): Dask 实现引入了一个附加参数 method。
39: 不支持 axis 参数。
40: 不支持 casting 参数。

dask.array.utils.meta_from_array

Bag