X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=mjo%2Fcone%2Frearrangement.py;fp=mjo%2Fcone%2Frearrangement.py;h=3e09c8ad948a8c70135809b6e3c5e5328babfa36;hb=6e240259a6453b592c136242b4c1738c24c35aed;hp=da6c595b29a39c79b8c44f90178a9ccf52c0174d;hpb=f6548daafb1a5c6352007d7518452169fcd84cf1;p=sage.d.git

diff --git a/mjo/cone/rearrangement.py b/mjo/cone/rearrangement.py
index da6c595..3e09c8a 100644
--- a/mjo/cone/rearrangement.py
+++ b/mjo/cone/rearrangement.py
@@ -169,10 +169,7 @@ def rearrangement_cone(p,n,lattice=None):
                          %
                          (lattice.rank(), n))
 
-    def d(j):
-        v = [1]*n    # Create the list of all ones...
-        v[j] = 1 - p # Now "fix" the ``j``th entry.
-        return v
-
-    G = identity_matrix(ZZ,n).rows() + [ d(j) for j in xrange(n) ]
+    I = identity_matrix(ZZ,n)
+    M = matrix.ones(ZZ,n) - p*I
+    G = identity_matrix(ZZ,n).rows() + M.rows()
     return Cone(G, lattice=lattice)