如何使用变量导入nltk.corpus中的指定语料库？

你遇到的问题很典型——Python的import语句是静态解析的，它会把你写的corp当作字面量名称去查找，而不是读取变量corp的值，所以才会报错找不到名为"corp"的语料库。

这里有两种简洁的解决方法，都能实现用变量指定语料库的需求：

方法一：使用getattr直接获取语料库对象 #

NLTK的corpus模块已经把所有可用的语料库作为自身的属性存储了，所以我们可以用getattr()函数，通过变量名动态获取对应的语料库：

from nltk import corpus

def chi_square(w1, w2, corp_name):
    # 动态获取指定的语料库
    corp = getattr(corpus, corp_name)
    # 接下来就可以像使用普通语料库一样操作corp了
    # 示例：获取语料库中的词列表
    words = corp.words()
    # 这里编写你的卡方检验逻辑
    ...

调用函数时，直接传入语料库的字符串名称即可，比如chi_square("apple", "banana", "brown")（假设你要导入brown语料库）。

方法二：使用importlib动态导入模块 #

如果需要更通用的动态导入方案（不止适用于NLTK），可以用Python标准库的importlib模块：

import importlib

def chi_square(w1, w2, corp_name):
    # 动态导入指定的语料库模块
    corp = importlib.import_module(f'nltk.corpus.{corp_name}')
    # 后续操作和方法一一致
    words = corp.words()
    ...

这两种方法都能完美解决你的问题，其中方法一更贴合NLTK的使用场景，代码也更简洁。

内容的提问来源于stack exchange，提问作者Peter Doyle