haojin2 commented on a change in pull request #17084: [numpy] add op median
URL: https://github.com/apache/incubator-mxnet/pull/17084#discussion_r380002267
##########
File path: python/mxnet/ndarray/numpy/_op.py
##########
@@ -6550,6 +6550,57 @@ def percentile(a, q, axis=None, out=None,
overwrite_input=None, interpolation='l
keepdims=keepdims, q_scalar=None, out=out)
+@set_module('mxnet.ndarray.numpy')
+def median(a, axis=None, out=None, overwrite_input=None, keepdims=False):
+ r"""
+ Compute the median along the specified axis.
+ Returns the median of the array elements.
+ Parameters
+ ----------
+ a : array_like
+ Input array or object that can be converted to an array.
+ axis : {int, sequence of int, None}, optional
+ Axis or axes along which the medians are computed. The default
+ is to compute the median along a flattened version of the array.
+ A sequence of axes is supported since version 1.9.0.
+ out : ndarray, optional
+ Alternative output array in which to place the result. It must
+ have the same shape and buffer length as the expected output,
+ but the type (of the output) will be cast if necessary.
+ keepdims : bool, optional
+ If this is set to True, the axes which are reduced are left
+ in the result as dimensions with size one. With this option,
+ the result will broadcast correctly against the original `arr`.
+ Returns
+ -------
+ median : ndarray
+ A new array holding the result. If the input contains integers
+ or floats smaller than ``float32``, then the output data-type is
+ ``np.float32``. Otherwise, the data-type of the output is the
+ same as that of the input. If `out` is specified, that array is
+ returned instead.
+ See Also
+ --------
+ mean, percentile
+ Examples
+ --------
+ >>> a = np.array([[10, 7, 4], [3, 2, 1]])
+ >>> a
+ array([[10, 7, 4],
+ [ 3, 2, 1]])
+ >>> np.median(a)
+ 3.5
+ >>> np.median(a, axis=0)
+ array([6.5, 4.5, 2.5])
+ >>> np.median(a, axis=1)
+ array([7., 2.])
+ """
+ from mxnet import np, npx
Review comment:
no need for the import here.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
