14 lines
848 B
JavaScript
14 lines
848 B
JavaScript
|
/* 最小 .npy 写出（float32 / int32，C 序） */
|
|||
|
|
'use strict';
|
|||
|
|
const fs = require('fs');
|
|||
|
|
function writeNpy(file, arr, shape, dtype) {
|
|||
|
|
const descr = dtype === 'int32' ? '<i4' : '<f4';
|
|||
|
|
let hdr = `{'descr': '${descr}', 'fortran_order': False, 'shape': (${shape.join(', ')}${shape.length === 1 ? ',' : ''}), }`;
|
|||
|
|
const pre = 10, total = Math.ceil((pre + hdr.length + 1) / 64) * 64;
|
|||
|
|
hdr = hdr + ' '.repeat(total - pre - hdr.length - 1) + '\n';
|
|||
|
|
const h = Buffer.alloc(pre); h.write('\x93NUMPY', 0, 'latin1'); h[6] = 1; h[7] = 0; h.writeUInt16LE(hdr.length, 8);
|
|||
|
|
const ta = dtype === 'int32' ? Int32Array.from(arr) : (arr instanceof Float32Array ? arr : Float32Array.from(arr));
|
|||
|
|
fs.writeFileSync(file, Buffer.concat([h, Buffer.from(hdr, 'latin1'), Buffer.from(ta.buffer, ta.byteOffset, ta.byteLength)]));
|
|||
|
|
}
|
|||
|
|
module.exports = { writeNpy };
|