Merge pull request #42 from BastienTr/fix41

Fix #41

Author: veeresht

commpy/channelcoding/interleavers.py

@@ -1,15 +1,15 @@
 
-
 # Authors: Veeresh Taranalli <veeresht@gmail.com>
 # License: BSD 3-Clause
 
 """ Interleavers and De-interleavers """
 
-from numpy import array, arange, zeros
+from numpy import arange, zeros
 from numpy.random import mtrand
 
 __all__ = ['RandInterlv']
 
+
 class _Interleaver:
 
     def interlv(self, in_array):
@@ -26,7 +26,7 @@ def interlv(self, in_array):
             Interleaved output data.
 
         """
-        out_array = array(map(lambda x: in_array[x], self.p_array))
+        out_array = in_array[self.p_array]
         return out_array
 
     def deinterlv(self, in_array):
@@ -48,6 +48,7 @@ def deinterlv(self, in_array):
             out_array[element] = in_array[index]
         return out_array
 
+
 class RandInterlv(_Interleaver):
     """ Random Interleaver.
 
@@ -76,9 +77,3 @@ class RandInterlv(_Interleaver):
     def __init__(self, length, seed):
         rand_gen = mtrand.RandomState(seed)
         self.p_array = rand_gen.permutation(arange(length))
-
-
-#class SRandInterlv(_Interleaver):
-
-
-#class QPPInterlv(_Interleaver):