Python 小技巧：发掘 args 和 kwargs 的乐趣

在 Python 中，args 和 kwargs 是函数调用的两个强大特性，它们允许您将任意数量的位置参数和参数传递给函数。了解如何有效地使用这些特性，可以极大地增强您的代码的灵活性和可读性。

args 和 kwargs 的基础

• args： args 是一个元组，它收集了传递给函数的所有位置参数。
• kwargs： kwargs 是一个字典，它包含了所有关键字参数。

让我们通过一个简单的示例来了解这些概念：

def my_function(x, y, *args, **kwargs):
    print("位置参数：", x, y)
    print("可变长度参数：", args)
    print("关键字参数：", kwargs)

当您调用此函数时，位置参数会传递给 x 和 y，而可变长度参数（如果存在）会传递给 args 元组。同样，关键字参数（如果存在）会传递给 kwargs 字典。

可变长度参数的优点

可变长度参数最明显的优点是它们允许您编写接受任意数量参数的函数。例如，以下函数计算任意数量输入数字的总和：

def sum_numbers(*nums):
    total = 0
    for num in nums:
        total += num
    return total

关键字参数的优点

关键字参数使您可以使用关键字来指定参数，而不是按位置。这可以提高代码的可读性和可维护性。例如，以下函数使用关键字参数设置文本文件的文件名和模式：

def write_to_file(filename, mode="w"):
    with open(filename, mode) as f:
        f.write("Hello, world!")

使用 args 和 kwargs 的提示

以下是使用 args 和 kwargs 的一些提示：

• 始终将 args 和 kwargs 放置在函数定义中的参数列表的最后。
• 使用 * 和 ** 符号来指示可变长度参数和关键字参数。
• 在函数文档中记录 args 和 kwargs 的预期用法。
• 使用默认参数值来简化函数调用。

示例

让我们看看一个使用 args 和 kwargs 的实际示例。我们创建一个函数，它计算两个或更多输入数字的平均值，并使用关键字参数指定结果精度：

def calculate_average(*numbers, precision=2):
    total = sum(numbers)
    average = total / len(numbers)
    return round(average, precision)

这个函数可以如下调用：

average = calculate_average(10, 20, 30)  # 平均值精确到小数点后 0 位
average = calculate_average(10, 20, 30, precision=3)  # 平均值精确到小数点后 3 位