在函数参数中列出理解

Question

In Python 2.7.1, I'm trying to provide a list of messages as the first argument, and a list of colors as the second argument. I want the second argument to default to a list of whites if it's not provided. This is the way I tried to do it:

def multicolor_message(msgs, colors=[libtcod.white for x in len(msgs)]):
#function body

libtcod.white is a part of the library I'm using and is in no way causing any problems. The compiler says the variable msgs is not defined. Obviously the msgs variable doesn't exist in this scope, but I need to create a list of appropriate length and assign it to colors . What is the cleanest way to do this?

Answer 1

I would do it like this:

def multicolor_message(msgs, colors=None):
  if colors is None:
    colors=[libtcod.white for x in len(msgs)]

Answer 2

That's not possible is python, as function's default arguments are executed at function definition time and your msgs variable will not be available until the function is called.

From the docs :

Default parameter values are evaluated when the function definition is executed. This means that the expression is evaluated once, when the function is defined, and that the same “pre-computed” value is used for each call. This is especially important to understand when a default parameter is a mutable object, such as a list or a dictionary: if the function modifies the object (eg by appending an item to a list), the default value is in effect modified. This is generally not what was intended. A way around this is to use None as the default, and explicitly test for it in the body of the function, eg:

def whats_on_the_telly(penguin=None):
    if penguin is None:
        penguin = []
    penguin.append("property of the zoo")
    return penguin