Skip to content

Commit e05d710

Browse files
authored
Update ndarray.js
Signed-off-by: Kaushikgtm <[email protected]>
1 parent 6cee644 commit e05d710

File tree

1 file changed

+7
-17
lines changed
  • lib/node_modules/@stdlib/stats/base/nanmskmin/lib

1 file changed

+7
-17
lines changed

lib/node_modules/@stdlib/stats/base/nanmskmin/lib/ndarray.js

Lines changed: 7 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -70,19 +70,6 @@ function nanmskmin( N, x, strideX, offsetX, mask, strideMask, offsetMask ) {
7070
ix = offsetX;
7171
im = offsetMask;
7272

73-
// Check for NaN in unmasked elements
74-
for ( i = 0; i < N; i++ ) {
75-
if ( mask[ im ] === 0 && isnan( x[ ix ] ) ) {
76-
return NaN;
77-
}
78-
ix += strideX;
79-
im += strideMask;
80-
}
81-
82-
// Reset indices
83-
ix = offsetX;
84-
im = offsetMask;
85-
8673
// Find the first unmasked element
8774
for ( i = 0; i < N; i++ ) {
8875
if ( mask[im] === 0 ) {
@@ -97,17 +84,20 @@ function nanmskmin( N, x, strideX, offsetX, mask, strideMask, offsetMask ) {
9784
}
9885

9986
min = x[ ix ];
100-
i += 1;
101-
if( isnan( min ) ) {
87+
if ( isnan( min ) ) {
10288
return min;
10389
}
104-
for ( i; i < N; i++ ) {
90+
91+
for ( i += 1; i < N; i++ ) {
10592
ix += strideX;
10693
im += strideMask;
107-
if ( mask[ im ] === 0 ) {
94+
if ( mask[ im ] !== 0 ) {
10895
continue;
10996
}
11097
v = x[ ix ];
98+
if ( isnan( v ) ) {
99+
return v;
100+
}
111101
if ( v < min || ( v === min && isNegativeZero( v ) ) ) {
112102
min = v;
113103
}

0 commit comments

Comments
 (0)