Блог

У меня есть файл, с которым я работаю. Я хочу создать функцию, которая считывает файлы и помещает содержимое в словарь. Затем этот словарь необходимо передать через основную функцию. Вот основная программа. Она не может быть изменена. Все, что я делаю, должно работать с основной программой.

 def main():
    sunspot_dict = {}
    file_str = raw_input("Open what data file: ")
    keep_going = True
    while keep_going:
        try:
            init_dictionary(file_str, sunspot_dict)
        except IOError:
            print "Data file error, try again"
            file_str = raw_input("Open what data file: ")    
            continue
        print "Jan, 1900-1905:", avg_sunspot(sunspot_dict, (1900,1905),(1,1))
        print "Jan-June, 2000-2011:", avg_sunspot(sunspot_dict, (2000,2011), (1,6))
        print "All years, Jan:", avg_sunspot(sunspot_dict, month_tuple=(1,1))
        print "All months, 1900-1905:", avg_sunspot(sunspot_dict, year_tuple=(1900,1905))
        try:
            print "Bad Year Key example:", avg_sunspot(sunspot_dict, (100,1000), (1,1))
        except KeyError:
            print "Bad Key raised"
        try:
            print "Bad Month Index example:", avg_sunspot(sunspot_dict, (2000,2011), (1,100))
        except IndexError:
            print "Bad Range raised"
        keep_going = False
    print "Main function finished normally."

    print sunspot_dict
  

Вот что у меня есть на данный момент:

 def init_dictionary(file_str, sunspot_dict):
    line = open(file_str, "rU")
    sunspot_dict = {}
    for i in line:
        sunspot_dict[i]
print sunspot_dict
  

Основываясь на вашем втором комментарии и коде предварительного просмотра, вы на правильном пути. Словари в Python передаются по ссылке, поэтому вы должны просто иметь возможность заполнить словарь, предоставленный в init_dictionary , и значения будут доступны в main() . В вашем случае вы создаете новый словарь в init_dictionary с помощью строки sunspot_dict = {} . Вы не хотите этого делать, потому что словарь, который вы хотите использовать, уже был создан в main() .

На этом этапе нам нужно было бы знать формат файла, который вы читаете. По сути, вам нужно открыть файл, затем, вероятно, прочитать его построчно, разбирая строки на пары ключ / значение и заполняя sunspot_dict .

Учитывая ограничение, которое вы даете в своем ответе Raymond, что вы не можете изменять init_dictionary() метод и что для этого требуется, чтобы вы передали ему словарь, который он затем заполнит, я рекомендую вам сделать свой собственный класс подклассом dict .

 class MySunspotDict(dict):
    def __init__(self, file_str):
        init_dictionary(file_str, self)

    # ... define your other methods here

sunspot_dict = MySunspotDict(file_str)
  

sunspot_dict = MySunspotDict(file_str) Строка идет туда, где у вас есть в данный момент init_dictionary(file_str, sunspot_dict) , поэтому ваша существующая file_str информация передается MySunspotDict конструктору ( __init__() методу), который затем передаст ее init_dictionary() .

init_dictionary() Вызов работает так же, как тот, который у вас уже есть, за исключением того, что он передает новый объект, а не пустой dict , функции, которая заполняет словарь. Поскольку ваш новый класс является производным от dict , то есть это подкласс, для этой цели он работает точно так же, как обычный словарь. Итак, у вас есть в основном dict , к которому вы можете присоединить свои собственные методы. (Вы также можете переопределить некоторые стандартные действия словаря, определив методы со специальными именами, но в данном случае вам не нужно этого делать.)

Обратите внимание, что в ваших методах self будет словарь — вам не нужно создавать словарь в вашем __init__() и сохранять его как атрибут вашего класса (вы увидите этот шаблон во многих кодах Python). Словарь уже существует ко времени __init__() выполнения.