From ae3664396878000dd5cbb82f2e210a91121adce0 Mon Sep 17 00:00:00 2001
From: Kenneth Harris <kenneth.harris@ucl.ac.uk>
Date: Thu, 29 May 2025 18:31:44 +0100
Subject: [PATCH] Added writeNPYshaped which lets you save with a prespecified
 shape

---
 constructNPYheader.m |  2 +-
 writeNPYshaped.m     | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+), 1 deletion(-)
 create mode 100644 writeNPYshaped.m

diff --git a/constructNPYheader.m b/constructNPYheader.m
index ca2840e..37e8c09 100644
--- a/constructNPYheader.m
+++ b/constructNPYheader.m
@@ -63,7 +63,7 @@
 
     for s = 1:length(shape)
         dictString = [dictString num2str(shape(s))];
-        if s<length(shape)
+        if s<length(shape) || s==1
             dictString = [dictString ', '];
         end
     end
diff --git a/writeNPYshaped.m b/writeNPYshaped.m
new file mode 100644
index 0000000..ad2ee97
--- /dev/null
+++ b/writeNPYshaped.m
@@ -0,0 +1,32 @@
+
+
+function writeNPYshaped(var, filename, shape)
+% function writeNPY(var, filename, shape)
+%
+% Only writes little endian, fortran (column-major) ordering; only writes
+% with NPY version number 1.0.
+%
+% shape is the size of the array as in the NPY file.  You need this to
+% override matlab's convention, e.g. (10, 1)rather than (10,).
+% if not specified, defaults to matlab's size(var)
+
+if nargin<3
+    shape = size(var);
+end
+
+if prod(shape)~=prod(size(var))
+    error('shape input does not matach number of elements in var')
+end
+
+dataType = class(var);
+
+header = constructNPYheader(dataType, shape);
+
+fid = fopen(filename, 'w');
+fwrite(fid, header, 'uint8');
+fwrite(fid, var, dataType);
+fclose(fid);
+
+
+end
+