|
7016
|
1 ## Copyright (C) 2000, 2001 Paul Kienzle |
|
|
2 ## |
|
|
3 ## This file is part of Octave. |
|
5165
|
4 ## |
|
7016
|
5 ## Octave is free software; you can redistribute it and/or modify it |
|
|
6 ## under the terms of the GNU General Public License as published by |
|
|
7 ## the Free Software Foundation; either version 3 of the License, or (at |
|
|
8 ## your option) any later version. |
|
5165
|
9 ## |
|
7016
|
10 ## Octave is distributed in the hope that it will be useful, but |
|
|
11 ## WITHOUT ANY WARRANTY; without even the implied warranty of |
|
|
12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
|
13 ## General Public License for more details. |
|
5165
|
14 ## |
|
|
15 ## You should have received a copy of the GNU General Public License |
|
7016
|
16 ## along with Octave; see the file COPYING. If not, see |
|
|
17 ## <http://www.gnu.org/licenses/>. |
|
5165
|
18 |
|
|
19 ## -*- texinfo -*- |
|
|
20 ## @deftypefn {Function File} {} unique (@var{x}) |
|
|
21 ## |
|
|
22 ## Return the unique elements of @var{x}, sorted in ascending order. |
|
|
23 ## If @var{x} is a row vector, return a row vector, but if @var{x} |
|
|
24 ## is a column vector or a matrix return a column vector. |
|
|
25 ## |
|
|
26 ## @deftypefnx {Function File} {} unique (@var{A}, 'rows') |
|
|
27 ## |
|
|
28 ## Return the unique rows of @var{A}, sorted in ascending order. |
|
|
29 ## |
|
|
30 ## @deftypefnx {Function File} {[@var{y}, @var{i}, @var{j}] = } unique (@var{x}) |
|
|
31 ## |
|
|
32 ## Return index vectors @var{i} and @var{j} such that @code{x(i)==y} and |
|
5532
|
33 ## @code{y(j)==x}. |
|
5642
|
34 ## @seealso{union, intersect, setdiff, setxor, ismember} |
|
5165
|
35 ## @end deftypefn |
|
|
36 |
|
|
37 function [y, i, j] = unique (x, r) |
|
|
38 |
|
5168
|
39 if (nargin < 1 || nargin > 2 || (nargin == 2 && ! strcmp (r, "rows"))) |
|
6046
|
40 print_usage (); |
|
5165
|
41 endif |
|
|
42 |
|
|
43 if (nargin == 1) |
|
6609
|
44 n = numel (x); |
|
5165
|
45 else |
|
5168
|
46 n = size (x, 1); |
|
5165
|
47 endif |
|
|
48 |
|
5168
|
49 y = x; |
|
5165
|
50 if (n < 1) |
|
|
51 i = j = []; |
|
5168
|
52 return; |
|
5165
|
53 elseif (n < 2) |
|
|
54 i = j = 1; |
|
5168
|
55 return; |
|
5165
|
56 endif |
|
|
57 |
|
5443
|
58 if (ischar (x)) |
|
5168
|
59 y = toascii (y); |
|
|
60 endif |
|
5165
|
61 |
|
5737
|
62 if (nargin == 2 && size (y, 2) > 1) |
|
5168
|
63 [y, i] = sortrows (y); |
|
5205
|
64 if (iscell (y)) |
|
|
65 match = cellfun ("size", y(1:n-1,:), 1) == cellfun ("size", y(2:n,:), 1); |
|
|
66 idx = find (match); |
|
|
67 match(idx) = all (char (y(idx)) == char (y(idx+1)), 2); |
|
|
68 match = all (match'); |
|
|
69 else |
|
|
70 match = all ([y(1:n-1,:) == y(2:n,:)]'); |
|
|
71 endif |
|
5165
|
72 idx = find (match); |
|
5168
|
73 y(idx,:) = []; |
|
5165
|
74 else |
|
5168
|
75 if (size (y, 1) != 1) |
|
|
76 y = y(:); |
|
|
77 endif |
|
|
78 [y, i] = sort (y); |
|
5205
|
79 if (iscell (y)) |
|
|
80 match = cellfun ("length", y(1:n-1)) == cellfun ("length", y(2:n)); |
|
|
81 idx = find(match); |
|
|
82 match(idx) = all (char (y(idx)) == char (y(idx+1)), 2); |
|
|
83 else |
|
|
84 match = [y(1:n-1) == y(2:n)]; |
|
|
85 endif |
|
5165
|
86 idx = find (match); |
|
5168
|
87 y(idx) = []; |
|
5165
|
88 endif |
|
|
89 |
|
|
90 ## I don't know why anyone would need reverse indices, but it |
|
|
91 ## was an interesting challenge. I welcome cleaner solutions. |
|
|
92 if (nargout >= 3) |
|
|
93 j = i; |
|
5168
|
94 j(i) = cumsum (prepad (! match, n, 1)); |
|
5165
|
95 endif |
|
5168
|
96 i(idx) = []; |
|
5165
|
97 |
|
5443
|
98 if (ischar (x)) |
|
|
99 y = char (y); |
|
5168
|
100 endif |
|
5165
|
101 |
|
|
102 endfunction |
|
|
103 |
|
|
104 %!assert(unique([1 1 2; 1 2 1; 1 1 2]),[1;2]) |
|
|
105 %!assert(unique([1 1 2; 1 0 1; 1 1 2],'rows'),[1 0 1; 1 1 2]) |
|
|
106 %!assert(unique([]),[]) |
|
|
107 %!assert(unique([1]),[1]) |
|
|
108 %!assert(unique([1 2]),[1 2]) |
|
|
109 %!assert(unique([1;2]),[1;2]) |
|
|
110 %!assert(unique([1,NaN,Inf,NaN,Inf]),[1,Inf,NaN,NaN]) |
|
5205
|
111 %!assert(unique({'Foo','Bar','Foo'}),{'Bar','Foo'}) |
|
|
112 %!assert(unique({'Foo','Bar','FooBar'}),{'Bar','Foo','FooBar'}) |
