numo-narray-alt 0.9.6 → 0.9.8

data/lib/numo/narray/extra.rb

@@ -1045,6 +1045,83 @@ module Numo
       a
     end
 
+    # Returns an index array of sort result.
+    #
+    # @example
+    #   require 'numo/narray'
+    #
+    #   a = Numo::DFloat[[0.1, 0.7],
+    #                    [0.4, 0.2],
+    #                    [0.2, 0.5]]
+    #   pp a.argsort
+    #   # =>
+    #   # Numo::Int32#shape=[3,2]
+    #   # [[0, 1],
+    #   #  [1, 0],
+    #   #  [0, 1]]
+    #   pp a.argsort(axis: 0)
+    #   # =>
+    #   # Numo::Int32#shape=[3,2]
+    #   # [[0, 1],
+    #   #  [2, 2],
+    #   #  [1, 0]]
+    #   pp a.argsort(axis: 1)
+    #   # =>
+    #   # Numo::Int32#shape=[3,2]
+    #   # [[0, 1],
+    #   #  [1, 0],
+    #   #  [0, 1]]
+    #   pp a.argsort(axis: nil)
+    #   # =>
+    #   # Numo::Int32#shape=[6]
+    #   # [0, 3, 4, 2, 5, 1]
+    #
+    # @overload argsort(axis: -1)
+    #   @param axis [Integer, nil] Axis along which to sort. Default is -1 (the last axis).
+    #     If `nil` is given, the array is flattened before sorting.
+    #   @return [Numo::Int32] An array of indices that would sort the array.
+    def argsort(axis_ = 'none', axis: -1)
+      raise NotImplementedError, "argsort is not implemented for #{self.class}" unless respond_to?(:sort_index)
+
+      axis = axis_ unless axis_ == 'none'
+
+      return flatten.sort_index if axis.nil?
+
+      axis = ndim + axis if axis.negative?
+      raise Numo::NArray::DimensionError, 'dimension is out of range' if axis.negative? || axis >= ndim
+
+      case ndim
+      when 1
+        sort_index
+      when 2
+        case axis
+        when 0
+          indices = transpose.sort_index(1)
+          indices.transpose - indices.min(1)
+        when 1
+          indices = sort_index(1)
+          indices - indices.min(1).expand_dims(1)
+        end
+      else
+        res = Numo::Int32.zeros(*shape)
+        slicer = Array.new(ndim)
+        slicer[axis] = true
+        other_axes = Array.new(ndim) { |i| i } - [axis]
+        axis_ids = other_axes.map do |d|
+          Array.new(shape[d]) { |i| i }
+        end
+        axis_ids.inject(:product).each do |indices|
+          indices = indices.flatten
+          other_axes.each_with_index do |d, i|
+            slicer[d] = indices[i]
+          end
+          sorted_indices = self[*slicer].sort_index
+          res[*slicer] = sorted_indices - sorted_indices.min
+        end
+        res
+      end
+    end
+
     # Return the sum along diagonals of the array.
     #
     # If 2-D array, computes the summation along its diagonal with the

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: numo-narray-alt
 version: !ruby/object:Gem::Version
-  version: 0.9.6
+  version: 0.9.8
 platform: ruby
 authors:
 - yoshoku
@@ -63,6 +63,36 @@ files:
 - ext/numo/narray/numo/types/uint_macro.h
 - ext/numo/narray/numo/types/xint_macro.h
 - ext/numo/narray/rand.c
+- ext/numo/narray/src/mh/math/acos.h
+- ext/numo/narray/src/mh/math/acosh.h
+- ext/numo/narray/src/mh/math/asin.h
+- ext/numo/narray/src/mh/math/asinh.h
+- ext/numo/narray/src/mh/math/atan.h
+- ext/numo/narray/src/mh/math/atan2.h
+- ext/numo/narray/src/mh/math/atanh.h
+- ext/numo/narray/src/mh/math/cbrt.h
+- ext/numo/narray/src/mh/math/cos.h
+- ext/numo/narray/src/mh/math/cosh.h
+- ext/numo/narray/src/mh/math/erf.h
+- ext/numo/narray/src/mh/math/erfc.h
+- ext/numo/narray/src/mh/math/exp.h
+- ext/numo/narray/src/mh/math/exp10.h
+- ext/numo/narray/src/mh/math/exp2.h
+- ext/numo/narray/src/mh/math/expm1.h
+- ext/numo/narray/src/mh/math/frexp.h
+- ext/numo/narray/src/mh/math/hypot.h
+- ext/numo/narray/src/mh/math/ldexp.h
+- ext/numo/narray/src/mh/math/log.h
+- ext/numo/narray/src/mh/math/log10.h
+- ext/numo/narray/src/mh/math/log1p.h
+- ext/numo/narray/src/mh/math/log2.h
+- ext/numo/narray/src/mh/math/sin.h
+- ext/numo/narray/src/mh/math/sinc.h
+- ext/numo/narray/src/mh/math/sinh.h
+- ext/numo/narray/src/mh/math/sqrt.h
+- ext/numo/narray/src/mh/math/tan.h
+- ext/numo/narray/src/mh/math/tanh.h
+- ext/numo/narray/src/mh/math/unary_func.h
 - ext/numo/narray/src/mh/mean.h
 - ext/numo/narray/src/mh/rms.h
 - ext/numo/narray/src/mh/stddev.h
@@ -93,7 +123,7 @@ metadata:
   homepage_uri: https://github.com/yoshoku/numo-narray-alt
   source_code_uri: https://github.com/yoshoku/numo-narray-alt
   changelog_uri: https://github.com/yoshoku/numo-narray-alt/blob/main/CHANGELOG.md
-  documentation_uri: https://gemdocs.org/gems/numo-narray-alt/0.9.6/
+  documentation_uri: https://gemdocs.org/gems/numo-narray-alt/0.9.8/
   rubygems_mfa_required: 'true'
 rdoc_options: []
 require_paths:
@@ -109,7 +139,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 3.7.2
 specification_version: 4
 summary: Numo::NArray Alternative is an experimental project forked from Numo::NArray.
 test_files: []