作为列表元素的函数

Question

我有以下代码。我想尝试不同的特征选择算法而不重复两次相同的代码，所以我将函数名称放在列表中并编写了以下代码以查看它是否有效。确实如此。

我的问题是，列表如何将函数名称作为其项，以及它在 for 循环中的实际工作方式？

from sklearn.datasets import load_digits
from sklearn.feature_selection import SelectKBest, chi2
from sklearn.feature_selection import mutual_info_classif
X, y = load_digits(return_X_y=True)

list=[mutual_info_classif,chi2]

for i in list:
    print(type(i))
    X_new = SelectKBest(i, k=20).fit_transform(X, y)
    print(X_new)
    print('hello')

*输出**

<class 'function'>
[[ 5. 13. 15. ...  6.  0.  0.]
 [ 0.  0.  9. ...  0. 10.  0.]
 [ 0.  3. 14. ...  0. 16.  9.]
 ...
 [ 1. 13.  2. ...  2.  6.  0.]
 [ 2. 14. 15. ...  5. 12.  0.]
 [10. 16.  1. ...  8. 12.  1.]]
hello
<class 'function'>
[[ 1.  0. 15. ...  6.  0.  0.]
 [ 5.  0.  9. ...  0. 10.  0.]
 [12.  0. 14. ...  0. 16.  9.]
 ...
 [ 1.  0.  2. ...  2.  6.  0.]
 [ 0.  0. 15. ...  5. 12.  0.]
 [ 1.  0.  1. ...  8. 12.  1.]]
hello

Answer 1

您没有将函数的“名称”这样放入列表中。函数是对象，就像 Python 中的其他一切一样。当您使用def 语句创建函数对象时，您将其绑定到一个名称，就像使用= 一样，并且还设置对象的__name__ 属性。当您将这些名称放在一个列表中时，该列表包含对您以后可以调用的函数对象的引用。

for 循环将引用一一绑定到名称 i。除了 type(i) 之外，您可能还需要打印 i.__name__ 以查看每次迭代将哪个函数传递给 SelectKBest。

在 Python 中以这种方式处理函数是很常见的。函数和其他可调用对象可以放入列表、字典、元组以及几乎任何其他数据结构中。由于functions are hashable，它们甚至可以用作字典键或放在集合中。您还可以通过分配属性来为函数对象添加属性。