1

我希望我的函数根据调用参数返回多个字典。例如，如果我不带参数调用函数，它应该返回所有字典，否则如果我用参数列表调用它，它应该从函数返回相应的字典。显然，如果我在没有参数或一个参数的情况下调用，它工作正常，但我在使用多个值时遇到问题。以下示例显示了此问题。

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

怀疑：返回列表而不是字典：

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

2 回答 2

11

一个函数只能返回一次。您不能循环并尝试在循环的每次迭代中返回一些东西；第一个返回并结束函数。如果你想“返回多个值”，你真正要做的就是返回一个包含所有值的值，比如值的列表或值的元组。

此外，最好将您的字典放在字典中（sup dawg），而不是使用局部变量来命名它们。然后你可以用钥匙把它们挑出来。

这是一种返回所选字典列表的方法：

>>> 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}]
于 2012-08-05T07:29:06.813 回答
0

为什么不让他们像这样

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]

请注意，这还将返回一个项目列表给您。

于 2012-08-05T07:39:14.900 回答