Python dictionary creation with conditioned values added