Modificar Funciones en Python

Reconsidera el ejemplo con la información del país. ¿Qué sucede si el parámetro name proporcionado no se encuentra en el conjunto de datos?


              1234567891011121314
            
# Data
countries_dict = {'USA': (9629091, 331002651), 'Canada': (9984670, 37742154), 
                  'Germany': (357114, 83783942), 'Brazil': (8515767, 212559417), 
                  'India': (3166391, 1380004385)}

# Defining a function
def country_information(d, name):
  print('Country:', name)
  print('Area:', d[name][0], 'sq km')
  print('Population:', round(d[name][1]/1000000, 2), 'MM')

# Testing the function
country_information(countries_dict, 'USA')
country_information(countries_dict, 'Ukraine')

¿Podemos manejar esta situación? ¡Absolutamente, implementando declaraciones condicionales!


              1234567891011121314151617
            
# Data
countries_dict = {'USA': (9629091, 331002651), 'Canada': (9984670, 37742154), 
                  'Germany': (357114, 83783942), 'Brazil': (8515767, 212559417), 
                  'India': (3166391, 1380004385)}

# Modify our function
def country_information_mod(d, name):
  if name not in d.keys():
    print("There is no information about", name)
  else:
    print("Country:", name)
    print("Area:", d[name][0], 'sq km')
    print("Population:", round(d[name][1]/1000000, 2), 'mln')
        
# Testing the function
country_information_mod(countries_dict, "USA")
country_information_mod(countries_dict, "Ukraine")

Nota

El método d.keys() es un método de diccionario que devuelve una vista que contiene todas las claves del diccionario d. Aquí, se utiliza para verificar si el name proporcionado existe entre las claves del diccionario.

Como se demostró, el mensaje de error en este formato revisado es más amigable para el usuario. Aunque existen muchos otros errores potenciales, hay numerosos métodos para manejarlos.

¿Todo estuvo claro?

¡Gracias por tus comentarios!

Sección 6. Capítulo 9