Use list comprehension

2015-06-06 14:12:02 +07:00 · 2015-06-06 14:12:02 +07:00 · bbc2855cc5
parent a059afa9ce
commit bbc2855cc5
1 changed files with 1 additions and 3 deletions
--- a/relational/relation.py
+++ b/relational/relation.py
@ -165,9 +165,7 @@ class relation (object):
            raise Exception('Invalid attributes for projection')
        newt = relation()
        # Create the header
-        h = []
-        for i in ids:
-            h.append(self.header.attributes[i])
+        h = [self.header.attributes[i] for i in ids]
        newt.header = header(h)

        # Create the body