How To Write 2d Array In Python