Возвращение нескольких словарей из функции python на основе вызывающего аргумента

Question

Я хочу, чтобы моя функция возвращала несколько словарей на основе вызывающих аргументов. например если я вызываю функцию без аргумента, она должна возвращать все словари else, если я вызываю ее со списком аргументов, она должна возвращать соответствующие словари из этой функции. По-видимому, если я звоню без аргументов или одного arg, он работает нормально, но у меня проблема с использованием его с несколькими значениями. Следующий пример показывает эту проблему.

def mydef(arg1=None): 
    a = {'a1':1, 'a2':2} 
    b = {'b1':1, 'b2':2} 
    c = {'c1':1, 'c2':2} 
    d = {'d1':1, 'd2':2} 
    if arg1 is None: 
     return a,b,c,d 
    else: 
     for x in arg1: 
      if x == 'a': 
       return a 
      elif x == 'b': 
       return b 

w,x,y,z = mydef() 
print type(w) 
print w 

s,t = mydef(['a', 'b']) 
print type(s) 
print s 
print type(t) 
print t 

Сомнение: Списки возвращаются вместо dicts:

def mydef(args=None): 
    dicts = { 'a' :{'a1' : 1, 'a2' :2}, 'b' : {'b1' : 1, 'b2' :2}} 
    if args is None: 
     args = ['a', 'b'] 
    return [dicts[d] for d in args] 


x,y = mydef() 
type(x) 
>> type 'dict' 

type(y) 
>> type 'dict' 

x = mydef(['a']) 
type(x) 
>> type 'list' 

Answer 1

Функция получает только вернуть один раз. Вы не можете зацикливаться и пытаться вернуть что-то на каждой итерации цикла; первый возвращает и заканчивает функцию. Если вы хотите «вернуть несколько значений», то вам действительно нужно вернуть одно значение, содержащее их все, например список значений или кортеж значений.

Кроме того, было бы лучше поместить словари в словарь (sup dawg) вместо использования локальных переменных, чтобы назвать их. Затем вы можете просто выбрать их ключом.

Вот способ сделать это, который возвращает список выбранных dicts:

>>> def returnDicts(args=None): 
...  dicts = { 
...   'a': {'a1':1, 'a2':2}, 
...   'b': {'b1':1, 'b2':2}, 
...   'c': {'c1':1, 'c2':2}, 
...   'd': {'d1':1, 'd2':2} 
...  } 
...  if args is None: 
...   args = ['a', 'b', 'c', 'd'] 
...  return [dicts[d] for d in args] 
>>> returnDicts() 
[{'a1': 1, 'a2': 2}, 
{'b1': 1, 'b2': 2}, 
{'c1': 1, 'c2': 2}, 
{'d1': 1, 'd2': 2}] 
>>> returnDicts(['a']) 
[{'a1': 1, 'a2': 2}] 
>>> returnDicts(['a', 'b']) 
[{'a1': 1, 'a2': 2}, {'b1': 1, 'b2': 2}] 

Answer 2

почему бы не сделать их dicts из dicts как так

def mydicts(arg1=None): 
    dicter = {'a': {'a1':1, 'a2':2}, 
       'b': {'b1':1, 'b2':2}, 
       'c': {'c1':1, 'c2':2}, 
       'd': {'d1':1, 'd2':2}, 
       } 

    #if arg1 is None return all the dictionaries. 
    if arg1 is None: 
     arg1 = ['a', 'b', 'c', 'd'] 

    # Check if arg1 is a list and if not make it one 
    # Example you pass it a str or int 

    if not isinstance(arg1, list): 
     arg1 = [arg1] 

    return [dicter.get(x, {}) for x in arg1] 

Примечание, это также будет возвращать список предметов обратно к вам.