Mercurial > hg > octave-nkf
annotate scripts/set/ismember.m @ 10509:ddbd812d09aa
properly compress sparse matrices after assembly
author | Jaroslav Hajek <highegg@gmail.com> |
---|---|
date | Mon, 12 Apr 2010 12:57:44 +0200 |
parents | 5edee330d4cb |
children | a8ce6bdecce5 |
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 |
10088
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
74 function [tf, a_idx] = ismember (a, s, varargin) |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
75 |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
76 if (nargin < 2 || nargin > 3) |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
77 print_usage (); |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
78 endif |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
79 |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
80 [a, s] = validargs ("ismember", a, s, varargin{:}); |
5181 | 81 |
9345 | 82 if (nargin == 2) |
83 s = s(:); | |
84 ## We do it this way, because we expect the array to be often sorted. | |
85 if (issorted (s)) | |
86 is = []; | |
87 else | |
88 [s, is] = sort (s); | |
89 endif | |
9364 | 90 |
9924 | 91 ## sort out NaNs in table |
9364 | 92 if (isreal (s) && ! isempty (s) && isnan (s(end))) |
93 s = s(1:end - sum (isnan (s))); | |
94 endif | |
9345 | 95 |
96 if (nargout > 1) | |
97 a_idx = lookup (s, a, "m"); | |
98 tf = logical (a_idx); | |
99 if (! isempty (is)) | |
100 a_idx(tf) = is (a_idx(tf)); | |
7128 | 101 endif |
102 else | |
9345 | 103 tf = lookup (s, a, "b"); |
104 endif | |
105 | |
10088
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
106 else |
5edee330d4cb
better argument checking and handling in set functions
Jaroslav Hajek <highegg@gmail.com>
parents:
9924
diff
changeset
|
107 |
9735
4d03aff3add3
fix typo in aab4f2aa9ed9
Jaroslav Hajek <highegg@gmail.com>
parents:
9733
diff
changeset
|
108 if (isempty (a) || isempty (s)) |
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
109 tf = false (rows (a), 1); |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
110 a_idx = zeros (rows (a), 1); |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
111 else |
9345 | 112 |
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
113 ## 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
|
114 [xx, ii, jj] = unique ([a; s], "rows", "last"); |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
115 na = rows (a); |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
116 jj = ii(jj(1:na)); |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
117 tf = jj > na; |
9345 | 118 |
9733
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
119 if (nargout > 1) |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
120 a_idx = max (0, jj - na); |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
121 endif |
aab4f2aa9ed9
make ismember more matlab compatible
Jaroslav Hajek <highegg@gmail.com>
parents:
9364
diff
changeset
|
122 |
9345 | 123 endif |
5178 | 124 endif |
125 | |
7128 | 126 endfunction |
127 | |
6533 | 128 %!assert (ismember ({''}, {'abc', 'def'}), false); |
129 %!assert (ismember ('abc', {'abc', 'def'}), true); | |
130 %!assert (isempty (ismember ([], [1, 2])), true); | |
7077 | 131 %!assert (isempty (ismember ({}, {'a', 'b'})), true); |
7128 | 132 %!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
|
133 %!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
|
134 %!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
|
135 %!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
|
136 %!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
|
137 %!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
|
138 %!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
|
139 %!fail ('ismember ([1 2 3], [5 4 3 1], ''rows'')'); |
7128 | 140 %!assert (ismember ({'foo', 'bar'}, {'foobar'}), logical ([0, 0])); |
141 %!assert (ismember ({'foo'}, {'foobar'}), false); | |
142 %!assert (ismember ({'bar'}, {'foobar'}), false); | |
143 %!assert (ismember ({'bar'}, {'foobar', 'bar'}), true); | |
144 %!assert (ismember ({'foo', 'bar'}, {'foobar', 'bar'}), logical ([0, 1])); | |
145 %!assert (ismember ({'xfb', 'f', 'b'}, {'fb', 'b'}), logical ([0, 0, 1])); | |
146 %!assert (ismember ("1", "0123456789."), true); | |
147 | |
148 %!test | |
7652
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
149 %! [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
|
150 %! assert (result, logical ([0, 0])) |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
151 %! assert (a_idx, [0, 0]); |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
152 |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
153 %!test |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
154 %! [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
|
155 %! assert (result, logical ([])) |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
156 %! assert (a_idx, []); |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
157 |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
158 %!test |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
159 %! [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
|
160 %! assert (result, logical ([0, 0])) |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
161 %! assert (a_idx, [0, 0]); |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
162 |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
163 %!test |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
164 %! [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
|
165 %! assert (result, logical ([0, 0])) |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
166 %! assert (a_idx, [0, 0]); |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
167 |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
168 %!test |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
169 %! [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
|
170 %! assert (result, false) |
b5731e43283a
ismember: correctly size idx output for empty args
John W. Eaton <jwe@octave.org>
parents:
7129
diff
changeset
|
171 %! assert (a_idx, 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 ({}, {'a', 'b'}); |
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 |
7128 | 179 %! [result, a_idx] = ismember([1 2 3 4 5], [3]); |
180 %! assert (all (result == logical ([0 0 1 0 0])) && all (a_idx == [0 0 1 0 0])); | |
181 | |
182 %!test | |
183 %! [result, a_idx] = ismember([1 6], [1 2 3 4 5 1 6 1]); | |
9345 | 184 %! assert (all (result == logical ([1 1])) && a_idx(2) == 7); |
7128 | 185 |
186 %!test | |
187 %! [result, a_idx] = ismember ([3,10,1], [0,1,2,3,4,5,6,7,8,9]); | |
188 %! assert (all (result == logical ([1, 0, 1])) && all (a_idx == [4, 0, 2])); | |
189 | |
190 %!test | |
191 %! [result, a_idx] = ismember ("1.1", "0123456789.1"); | |
9364 | 192 %! assert (all (result == logical ([1, 1, 1])) && all (a_idx == [12, 11, 12])); |
7128 | 193 |
194 %!test | |
195 %! [result, a_idx] = ismember([1:3; 5:7; 4:6], [0:2; 1:3; 2:4; 3:5; 4:6], 'rows'); | |
196 %! assert (all (result == logical ([1; 0; 1])) && all (a_idx == [2; 0; 5])); | |
197 | |
198 %!test | |
199 %! [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'); | |
200 %! assert (all (result == logical ([1; 0; 1])) && all (a_idx == [1; 0; 2])); | |
201 |