how to get diamond pattern in python