How To Write A 2d Array Python