diff --git a/relational/relation.py b/relational/relation.py index e3f163d..3e8632f 100644 --- 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