这些代码怎么才能在运行完if elif 后继续运行for in？（本人刚入门）

def main() :
    while True :
        menu()
        choice = int(input('请选择'))
        d1 = {}
        d2 = {}
        if choice in [0,1] :
            if choice == 0 :
                d1 = file1()
            elif choice == 1 :
                d2 = file2()
            for key1 in d1.keys() :
                if key1 in d2.keys() :
                        d1[key1] = date_subtract(d1[key1], d2[key1])
            print('d1-d2差值：如下')
            print(d1)

先检查第13行，確定会成不

“for key1 ”后面的代码块，如果是需要在choice == 1时执行，就需要和d2 = file2()对齐，如果是执行完if判断再执行，就可以和if choice in [0,1] :对齐