### 使用 Python 提取核心药物组的函数设计 为了从大量的处方数据中提取核心药物组，可以采用基于关联规则挖掘的技术，例如 Apriori 或 FP-Growth 算法。这些方法可以帮助识别频繁项集以及它们之间的关系。以下是实现这一目标的一个可能方案： #### 1. 数据准备 假设输入是一个 DataFrame 表格 `prescription_data`，其中每一行代表一条处方记录，列名表示不同的药品名称，值为布尔型（True/False），指示某药是否存在于该处方中。 ```python import pandas as pd from mlxtend.frequent_patterns import apriori, association_rules def extract_core_drug_groups(prescription_data, min_support=0.1, metric="confidence", min_threshold=0.7): """ 提取核心药物组合 参数: prescription_data (pd.DataFrame): 输入的处方数据表，布尔型矩阵形式。 min_support (float): 支持度阈值，默认为0.1。 metric (str): 关联规则评估指标，默认为'confidence'。 min_threshold (float): 度量阈值，默认为0.7。 返回: core_drugs (pd.DataFrame): 核心药物组合及其支持度和置信度信息。 """ # 计算频繁项集 frequent_itemsets = apriori(prescription_data, min_support=min_support, use_colnames=True) # 基于频繁项集生成关联规则 rules = association_rules(frequent_itemsets, metric=metric, min_threshold=min_threshold) # 过滤出核心药物组（前件与后件） core_drugs = rules[(rules['antecedents'].apply(len) >= 2) & (rules['consequents'].apply(len) >= 1)] return core_drugs ``` 此函数利用了 `mlxtend` 库中的 `apriori` 和 `association_rules` 方法来计算频繁项集和支持度、置信度等指标[^5]。 #### 2. 示例代码运行 下面是如何调用上述函数并打印结果的例子： ```python # 构造示例数据 data = { 'DrugA': [True, False, True, True], 'DrugB': [True, True, False, True], 'DrugC': [False, True, True, True], 'DrugD': [True, False, True, False] } df_prescriptions = pd.DataFrame(data) # 调用函数 core_drug_groups = extract_core_drug_groups(df_prescriptions, min_support=0.2, metric="lift", min_threshold=1.2) print(core_drug_groups[['antecedents', 'consequents', 'support', 'confidence', 'lift']]) ``` 通过调整参数 `min_support`, `metric`, 和 `min_threshold` 可以控制筛选条件严格程度，从而获得更精确的核心药物组合列表[^6]。 --- ###