Модифікації Функцій

Повернімося до прикладу з інформацією про країну. Що станеться, якщо параметр name не буде знайдено в даних?


              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')

Чи можна впоратися з цією проблемою? Безумовно! За допомогою умовних операторів!


              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")

Примітка
d.keys() - це метод, який використовується зі словниками і повертає список усіх ключів у словнику (у даному випадку d). У цьому прикладі він використовується для перевірки, чи існує name серед ключів словника.

Як було продемонстровано, повідомлення про помилку в цьому оновленому форматі є більш зручним для користувача. Хоча існує багато інших потенційних помилок і методів їх усунення, вони виходять за рамки цього курсу.

Все було зрозуміло?

Секція 6. Розділ 9