呼び出し引数に基づいて、関数が複数の辞書を返すようにします。たとえば、引数なしで関数を呼び出すと、すべての辞書が返されます。引数のリストを指定して呼び出すと、関数から対応する辞書が返されます。どうやら、引数なしまたは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'