Golang Map Function In Python