Create 2 Dimensional Array Python Size Function