4.2.3 用函数增强程序的通用性

我们说过，程序 4.4 在减少重复代码和模块化两方面已经做得很好，但这并不意味着该 程序在各方面都已经完美。例如，如果我们希望换用字符"^"再画一棵树，以便比较哪个更 好看些，该如何做呢？显见的做法是仿照用"*"画树的代码重写画树冠的函数，而树干部分 可以重用。于是得到下面的代码：

【程序 4.5】tree5.py

def treetop1(): 
    print "   *" 
    print "  ***" 
    print " *****" 
    print "*******"
def treetop2(): 
    print "   ^" 
    print "  ^^^" 
    print " ^^^^^" 
    print "^^^^^^^"
def star_treetop(): 
    treetop1() 
    treetop1()
def caret_treetop(): 
    treetop2() 
    treetop2()
def treetrunk(): 
    print "   #"
    print "   #"
    print "   #"
def main():
    star_treetop() 
    treetrunk() 
    print 
    caret_treetop() 
    treetrunk()
main()

此版本的执行结果如下：

   *
  ***
 *****
*******
   *
  ***
 *****
*******
   #
   #
   #
   ^
  ^^^
 ^^^^^
^^^^^^^
   ^
  ^^^
 ^^^^^
^^^^^^^
   #
   #
   #

虽然程序 4.5 满足了功能需求，但是从程序设计角度说是很笨拙的，因为这是一种“头 痛医头脚痛医脚”的方法，即为每一种特殊情形创建新的代码。更好的做法是用一个一般的 函数来处理所有特殊情形。鉴于 treetop1 和 treetop2 的非常类似，我们可以从他们抽 象出一个通用的画树冠的函数，使得该函数能够取代 treetop1 和 treetop2。

函数的通用性可以通过引入参数（parameter）来实现。要理解参数的作用，可以简单地与数学函数的自变量进行类比。以函数 f(x) = x2 为例，对于给定的自变量值 10，函数计算 出函数值 f(10) = 100；换不同的自变量值 20，则函数又计算出另一个函数值 f(20) = 400。编 程语言中的函数参数具有类似的行为，输入不同的参数值，则函数执行后可产生不同的结果。

下面我们设计一个通用的画树冠的函数 treetop(ch)，其中参数 ch 表示用来作画的 字符。为了控制树的形状，函数定义中使用了字符串格式化运算。

>>> def treetop(ch):
        print " %s" % (ch) 
        print " %s" % (3 * ch) 
        print " %s" % (5 * ch) 
        print "%s" % (7 * ch)

在交互环境定义了函数 treetop(ch)后，我们接着来测试它的效果。下面是测试例子：

>>> treetop('*')
   *
  ***
 *****
*******
>>> treetop('^')
   ^
  ^^^
 ^^^^^
^^^^^^^
>>> treetop('A') 
   A
  AAA 
 AAAAA 
AAAAAAA

可见函数 treetop(ch)确实具有通用性，只要为它的参数提供一个字符值，就能用该字符 画出树冠形状。下面我们利用 treetop(ch)函数来改写程序 4.5：

【程序 4.6】tree6.py

def treetop(ch):
    print " %s" % (ch) 
    print " %s" % (3 * ch) 
    print " %s" % (5 * ch) 
    print "%s" % (7 * ch)
def star_treetop(): 
    treetop("*")
    treetop("*")
def caret_treetop(): 
    treetop("^")
    treetop("^")
def treetrunk(): 
    print " #"
    print " #"
    print " #"
def main():
    star_treetop() 
    treetrunk() 
    print 
    caret_treetop() 
    treetrunk()
main()

此版本的执行结果和程序 4.5 完全一样，但是比较两者的代码会发现，程序 4.6 将程序 4.5

中的两个函数合二为一，增强了通用性。以后如果想换用其他字符画树，修改程序 4.6 比修

改程序 4.5 要简单得多。