Mercurial > hg > octave-nkf
annotate scripts/set/ismember.m @ 9924:d0d6ff39db54
simplify ismember
| author | Jaroslav Hajek <highegg@gmail.com> |
|---|---|
| date | Sat, 05 Dec 2009 21:54:21 +0100 |
| parents | 4d03aff3add3 |
| children | 5edee330d4cb |
| rev | line source |
|---|---|
| 8920 | 1 ## Copyright (C) 2000, 2005, 2006, 2007, 2008, 2009 Paul Kienzle |
| 9345 | 2 ## Copyright (C) 2009 Jaroslav Hajek |
| 5178 | 3 ## |
| 5181 | 4 ## This file is part of Octave. |
| 5178 | 5 ## |
| 5181 | 6 ## Octave is free software; you can redistribute it and/or modify it |
| 7 ## under the terms of the GNU General Public License as published by | |
| 7016 | 8 ## the Free Software Foundation; either version 3 of the License, or (at |
| 9 ## your option) any later version. | |
| 5181 | 10 ## |
| 11 ## Octave is distributed in the hope that it will be useful, but | |
| 12 ## WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| 14 ## General Public License for more details. | |
| 5178 | 15 ## |
| 16 ## You should have received a copy of the GNU General Public License | |
| 7016 | 17 ## along with Octave; see the file COPYING. If not, see |
| 18 ## <http://www.gnu.org/licenses/>. | |
| 5178 | 19 |
| 20 ## -*- texinfo -*- | |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
21 ## @deftypefn {Function File} {[@var{tf} =} ismember (@var{A}, @var{S}) |
|
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
22 ## @deftypefnx {Function File} {[@var{tf}, @var{S_idx}] =} ismember (@var{A}, @var{S}) |
|
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
23 ## @deftypefnx {Function File} {[@var{tf}, @var{S_idx}] =} ismember (@var{A}, @var{S}, "rows") |
|
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
24 ## Return a matrix @var{tf} with the same shape as @var{A} which has a 1 if |
|
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
25 ## @code{A(i,j)} is in @var{S} and 0 if it is not. If a second output argument |
|
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
26 ## is requested, the index into @var{S} of each of the matching elements is |
| 7128 | 27 ## also returned. |
| 28 ## | |
| 29 ## @example | |
| 30 ## @group | |
| 31 ## a = [3, 10, 1]; | |
| 32 ## s = [0:9]; | |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
33 ## [tf, s_idx] = ismember (a, s); |
| 7128 | 34 ## @result{} tf = [1, 0, 1] |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
35 ## @result{} s_idx = [4, 0, 2] |
| 7128 | 36 ## @end group |
| 37 ## @end example | |
| 38 ## | |
| 39 ## The inputs, @var{A} and @var{S}, may also be cell arrays. | |
| 40 ## | |
| 41 ## @example | |
| 42 ## @group | |
| 43 ## a = @{'abc'@}; | |
| 44 ## s = @{'abc', 'def'@}; | |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
45 ## [tf, s_idx] = ismember (a, s); |
| 7128 | 46 ## @result{} tf = [1, 0] |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
47 ## @result{} s_idx = [1, 0] |
| 7128 | 48 ## @end group |
| 49 ## @end example | |
| 50 ## | |
| 7129 | 51 ## With the optional third argument @code{"rows"}, and matrices |
| 52 ## @var{A} and @var{S} with the same number of columns, compare rows in | |
| 53 ## @var{A} with the rows in @var{S}. | |
| 7128 | 54 ## |
| 55 ## @example | |
| 56 ## @group | |
| 57 ## a = [1:3; 5:7; 4:6]; | |
| 58 ## s = [0:2; 1:3; 2:4; 3:5; 4:6]; | |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
59 ## [tf, s_idx] = ismember(a, s, 'rows'); |
| 7128 | 60 ## @result{} tf = logical ([1; 0; 1]) |
|
9276
f5e4b5fd1f1e
Update ismember examples to remove incorrect reference to residue function
Rik <rdrider0-list@yahoo.com>
parents:
9051
diff
changeset
|
61 ## @result{} s_idx = [2; 0; 5]; |
| 7128 | 62 ## @end group |
| 63 ## @end example | |
| 64 ## | |
|
8286
6f2d95255911
fix @seealso references to point to existing anchors
Thorsten Meyer <thorsten.meyier@gmx.de>
parents:
7883
diff
changeset
|
65 ## @seealso{unique, union, intersect, setxor, setdiff} |
| 5178 | 66 ## @end deftypefn |
| 67 | |
| 7129 | 68 ## Author: Paul Kienzle <pkienzle@users.sf.net> |
| 69 ## Author: Søren Hauberg <hauberg@gmail.com> | |
| 70 ## Author: Ben Abbott <bpabbott@mac.com> | |
| 5181 | 71 ## Adapted-by: jwe |
| 9345 | 72 ## Reimplemented using lookup & unique: Jaroslav Hajek <highegg@gmail.com> |
| 5181 | 73 |
| 7128 | 74 function [tf, a_idx] = ismember (a, s, rows_opt) |
| 5181 | 75 |
| 9345 | 76 if (nargin == 2) |
| 77 ica = iscellstr (a); | |
| 78 ics = iscellstr (s); | |
| 79 if (ica || ics) | |
| 80 if (ica && ischar (s)) | |
| 81 s = cellstr (s); | |
| 82 elseif (ics && ischar (a)) | |
| 83 a = cellstr (a); | |
| 84 elseif (! (ica && ics)) | |
| 85 error ("ismember: invalid argument types"); | |
| 86 endif | |
| 87 elseif (! isa (a, class (s))) | |
| 88 error ("ismember: both input arguments must be the same type"); | |
| 89 elseif (! ischar (a) && ! isnumeric (a)) | |
| 90 error ("ismember: input arguments must be arrays, cell arrays, or strings"); | |
| 91 endif | |
| 92 | |
| 93 s = s(:); | |
| 94 ## We do it this way, because we expect the array to be often sorted. | |
| 95 if (issorted (s)) | |
| 96 is = []; | |
| 97 else | |
| 98 [s, is] = sort (s); | |
| 99 endif | |
| 9364 | 100 |
| 9924 | 101 ## sort out NaNs in table |
| 9364 | 102 if (isreal (s) && ! isempty (s) && isnan (s(end))) |
| 103 s = s(1:end - sum (isnan (s))); | |
| 104 endif | |
| 9345 | 105 |
| 106 if (nargout > 1) | |
| 107 a_idx = lookup (s, a, "m"); | |
| 108 tf = logical (a_idx); | |
| 109 if (! isempty (is)) | |
| 110 a_idx(tf) = is (a_idx(tf)); | |
| 7128 | 111 endif |
| 112 else | |
| 9345 | 113 tf = lookup (s, a, "b"); |
| 114 endif | |
| 115 | |
| 116 elseif (nargin == 3 && strcmpi (rows_opt, "rows")) | |
| 117 if (iscell (a) || iscell (s)) | |
| 118 error ("ismember: cells not supported with ""rows"""); | |
| 119 elseif (! isa (a, class (s))) | |
| 120 error ("ismember: both input arguments must be the same type"); | |
| 121 elseif (! ischar (a) && ! isnumeric (a)) | |
| 122 error ("ismember: input arguments must be arrays, cell arrays, or strings"); | |
| 7128 | 123 endif |
|
9735
4d03aff3add3
fix typo in aab4f2aa9ed9
Jaroslav Hajek <highegg@gmail.com>
parents:
9733
diff
changeset
|
124 if (isempty (a) || isempty (s)) |
|
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
125 tf = false (rows (a), 1); |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
126 a_idx = zeros (rows (a), 1); |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
127 elseif (columns (a) != columns (s)) |
| 9345 | 128 error ("ismember: number of columns must match"); |
|
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
129 else |
| 9345 | 130 |
|
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
131 ## FIXME: lookup does not support "rows", so we just use unique. |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
132 [xx, ii, jj] = unique ([a; s], "rows", "last"); |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
133 na = rows (a); |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
134 jj = ii(jj(1:na)); |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
135 tf = jj > na; |
| 9345 | 136 |
|
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
137 if (nargout > 1) |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
138 a_idx = max (0, jj - na); |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
139 endif |
|
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
140 |
| 9345 | 141 endif |
| 7128 | 142 else |
| 6046 | 143 print_usage (); |
| 5178 | 144 endif |
| 145 | |
| 7128 | 146 endfunction |
| 147 | |
| 6533 | 148 %!assert (ismember ({''}, {'abc', 'def'}), false); |
| 149 %!assert (ismember ('abc', {'abc', 'def'}), true); | |
| 150 %!assert (isempty (ismember ([], [1, 2])), true); | |
| 7077 | 151 %!assert (isempty (ismember ({}, {'a', 'b'})), true); |
| 7128 | 152 %!assert (ismember ('', {'abc', 'def'}), false); |
|
7883
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
153 %!fail ('ismember ([], {1, 2})'); |
|
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
154 %!fail ('ismember ({[]}, {1, 2})'); |
|
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
155 %!fail ('ismember ({}, {1, 2})'); |
|
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
156 %!fail ('ismember ({1}, {''1'', ''2''})'); |
|
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
157 %!fail ('ismember (1, ''abc'')'); |
|
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
158 %!fail ('ismember ({''1''}, {''1'', ''2''},''rows'')'); |
|
3092dd54ad95
fix expected output from lasterr in tests; fix fail tests in ismember.m
John W. Eaton <jwe@octave.org>
parents:
7652
diff
changeset
|
159 %!fail ('ismember ([1 2 3], [5 4 3 1], ''rows'')'); |
| 7128 | 160 %!assert (ismember ({'foo', 'bar'}, {'foobar'}), logical ([0, 0])); |
| 161 %!assert (ismember ({'foo'}, {'foobar'}), false); | |
| 162 %!assert (ismember ({'bar'}, {'foobar'}), false); | |
| 163 %!assert (ismember ({'bar'}, {'foobar', 'bar'}), true); | |
| 164 %!assert (ismember ({'foo', 'bar'}, {'foobar', 'bar'}), logical ([0, 1])); | |
| 165 %!assert (ismember ({'xfb', 'f', 'b'}, {'fb', 'b'}), logical ([0, 0, 1])); | |
| 166 %!assert (ismember ("1", "0123456789."), true); | |
| 167 | |
| 168 %!test | |
|
7652
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
169 %! [result, a_idx] = ismember ([1, 2], []); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
170 %! assert (result, logical ([0, 0])) |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
171 %! assert (a_idx, [0, 0]); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
172 |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
173 %!test |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
174 %! [result, a_idx] = ismember ([], [1, 2]); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
175 %! assert (result, logical ([])) |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
176 %! assert (a_idx, []); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
177 |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
178 %!test |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
179 %! [result, a_idx] = ismember ({'a', 'b'}, ''); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
180 %! assert (result, logical ([0, 0])) |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
181 %! assert (a_idx, [0, 0]); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
182 |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
183 %!test |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
184 %! [result, a_idx] = ismember ({'a', 'b'}, {}); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
185 %! assert (result, logical ([0, 0])) |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
186 %! assert (a_idx, [0, 0]); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
187 |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
188 %!test |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
189 %! [result, a_idx] = ismember ('', {'a', 'b'}); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
190 %! assert (result, false) |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
191 %! assert (a_idx, 0); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
192 |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
193 %!test |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
194 %! [result, a_idx] = ismember ({}, {'a', 'b'}); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
195 %! assert (result, logical ([])) |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
196 %! assert (a_idx, []); |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
197 |
|
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
198 %!test |
| 7128 | 199 %! [result, a_idx] = ismember([1 2 3 4 5], [3]); |
| 200 %! assert (all (result == logical ([0 0 1 0 0])) && all (a_idx == [0 0 1 0 0])); | |
| 201 | |
| 202 %!test | |
| 203 %! [result, a_idx] = ismember([1 6], [1 2 3 4 5 1 6 1]); | |
| 9345 | 204 %! assert (all (result == logical ([1 1])) && a_idx(2) == 7); |
| 7128 | 205 |
| 206 %!test | |
| 207 %! [result, a_idx] = ismember ([3,10,1], [0,1,2,3,4,5,6,7,8,9]); | |
| 208 %! assert (all (result == logical ([1, 0, 1])) && all (a_idx == [4, 0, 2])); | |
| 209 | |
| 210 %!test | |
| 211 %! [result, a_idx] = ismember ("1.1", "0123456789.1"); | |
| 9364 | 212 %! assert (all (result == logical ([1, 1, 1])) && all (a_idx == [12, 11, 12])); |
| 7128 | 213 |
| 214 %!test | |
| 215 %! [result, a_idx] = ismember([1:3; 5:7; 4:6], [0:2; 1:3; 2:4; 3:5; 4:6], 'rows'); | |
| 216 %! assert (all (result == logical ([1; 0; 1])) && all (a_idx == [2; 0; 5])); | |
| 217 | |
| 218 %!test | |
| 219 %! [result, a_idx] = ismember([1.1,1.2,1.3; 2.1,2.2,2.3; 10,11,12], [1.1,1.2,1.3; 10,11,12; 2.12,2.22,2.32], 'rows'); | |
| 220 %! assert (all (result == logical ([1; 0; 1])) && all (a_idx == [1; 0; 2])); | |
| 221 |