我将用一些非常简单的函数来说明我的问题,这些函数只是平方数。但我想了解的是编写函数的通用最佳实践。
假设我有一个要平方的数字,我可以编写以下函数。
如果我有一个数字列表,想要得到每个元素被平方后的列表,我可以使用:
假设我有一个要平方的数字,我可以编写以下函数。
def square (x):
y = x**2
return y
如果我有一个数字列表,想要得到每个元素被平方后的列表,我可以使用:
def square (x_list):
y_list = []
for x in x_list:
y_list.append( x**2 )
return y_list
但是我想写一个函数来处理这两种情况。它将查看输入是否为数字或列表,并相应地进行操作。我可以使用 type
来检测类型,但我想知道最pythonic的方法是什么。