From 4524f994e830cdf5bfd4989185471f80d562b869 Mon Sep 17 00:00:00 2001
From: Stefan van der Walt <stefan@sun.ac.za>
Date: Tue, 8 May 2012 16:21:53 -0700
Subject: [PATCH] ENH: Better formulation of _stackcopy.

---
 skimage/transform/_warp.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/skimage/transform/_warp.py b/skimage/transform/_warp.py
index 15c3d2ee..29a8531e 100644
--- a/skimage/transform/_warp.py
+++ b/skimage/transform/_warp.py
@@ -10,15 +10,19 @@ def _stackcopy(a, b):
 
       a[:,:,0] = a[:,:,1] = ... = b
 
+    Parameters
+    ----------
+    a : (M, N) or (M, N, P) ndarray
+        Target array.
+    b : (M, N)
+        Source array.
+
     Notes
     -----
     Color images are stored as an ``MxNx3`` or ``MxNx4`` arrays.
 
     """
-    if a.ndim == 3:
-        a.transpose().swapaxes(1, 2)[:] = b
-    else:
-        a[:] = b
+    a[:] = b[:, :, np.newaxis]
 
 def warp(image, reverse_map, map_args={},
          output_shape=None, order=1, mode='constant', cval=0.):