将必需的参数传递给可选的

Question

尝试制作一个棋盘，如果只有一个参数，则必须是正方形（如果有两个参数，则必须是矩形）。

代码示例：

class Game:
    def __init__(self,h,w=h): # lookie-lookie 
       self.board = [[0 for i in range(w)] for j in range(h)]

我的期望：

b = Game(2)
b.board
#should return [[0,0],[0,0]]
b = Game(2,1)
b.board
#should return [[0],[0]]

我得到了什么：

NameError: name 'h' is not defined

我应该如何解决它？

Answer 1

在函数定义时，对传递给可选参数的值进行评估，但是，在那一刻，h 没有定义。您可以改为使用标记值作为可选参数的占位符：

class Game(object):
    def __init__(self,h,w=None):
        if w is None:
           w = h

如果您使用的是 Python 2，请记住子类化 object。