Create 2d Array Python List Comprehension Syntax