mirror of
https://github.com/chanzuckerberg/cellxgene.git
synced 2026-09-21 02:48:11 +08:00
51 lines
1.3 KiB
JavaScript
51 lines
1.3 KiB
JavaScript
/*
|
|
Array range creation
|
|
|
|
range(start, stop, step) -> Array
|
|
This is identical to https://docs.python.org/3/library/functions.html#func-range
|
|
Returns new array filled with a range of numbers.
|
|
|
|
Usage:
|
|
|
|
range(stop) - start defaults to zero, step defaults to 1
|
|
range(start, stop, [step]) - step defaults to 1
|
|
|
|
Examples:
|
|
range(3) -> [0, 1, 2]
|
|
range(1, 3) -> [1, 2]
|
|
range(1, 5, 2) -> [1, 3]
|
|
|
|
|
|
rangeFill(array, start, step) -> array
|
|
Fill entire array with values, from start, by step. Returns first array.
|
|
start defaults to zero, step defaults to one.
|
|
|
|
*/
|
|
|
|
function _doFill(arr, start, step, count) {
|
|
for (let idx = 0, val = start; idx < count; idx += 1, val += step) {
|
|
arr[idx] = val;
|
|
}
|
|
return arr;
|
|
}
|
|
|
|
export function rangeFill(arr, start = 0, step = 1) {
|
|
return _doFill(arr, start, step, arr.length);
|
|
}
|
|
|
|
export function range(start, stop, step) {
|
|
if (start === undefined) return [];
|
|
if (stop === undefined) {
|
|
stop = start;
|
|
start = 0;
|
|
}
|
|
step = step || 1; // catch undefind and zero
|
|
const len = Math.max(Math.ceil((stop - start) / step), 0);
|
|
return _doFill(new Array(len), start, step, len);
|
|
}
|
|
|
|
export function linspace(start, stop, nsteps) {
|
|
const delta = (stop - start) / (nsteps - 1).toFixed();
|
|
return range(0, nsteps, 1).map((i) => start + i * delta);
|
|
}
|