How To Write A 2d Array In Python