从 Python 中的许多选项中进行选择的干净方法

Question

我从事数据科学工作，在清理 Pandas 数据框时遇到的一个典型问题是将列从一种字符串格式转换为另一种格式（特别是，我正在查看的字符串是化学标识符，它们中的每一个都代表一个分子一种晦涩难懂的方式，因此仅通过查看字符串就不容易理解）。我有许多小函数（继承自一个名为 RDKit 的化学库）来在格式之间进行转换，每个转换对（即输入格式和输出格式）大约有一个函数。函数名太多了，记不住了。我想编写一个包装函数，将所有这些聚合成一个更大的、具有简洁设计和用户界面的函数。

问题是：给定输入和输出格式，从许多可能的小型转换函数中选择什么是一种干净的方法？我应该使用存储小转换函数名称的字典吗？

例如，假设我想从“微笑”格式转换为“inchi keys”格式，我目前的操作如下：

from rdkit import Chem


def smile2inchikey(smile):
    mol = Chem.MolFromSmiles(smile)
    inchikey = Chem.inchi.MolToInchiKey(mol)
    return inchikey

我不想手动调用smile2inchikey（或Chem.MolFromSmiles和Chem.inchi.MolToInchiKey），而是想编写以下函数：

def fancy_multiconverter(input_string, input_format, output_format):
    pass

将input_string（以input_format 格式给出）返回为output_format 格式。

Answer 1

也许这也是@Quinten Cabo 的意思，但您可以使用每次转换为的单位。

然后您可以使用带有函数的字典来转换成这个单元和从这个单元转换出来：

convert_to_reference = {
                         "format1": function1,
                         "format2": function2,
                       }
convert_from_reference = {
                           ...
                         }
reference = convert_to_reference[input("input format: ")](input("input value: "))
output = convert_from_reference[input("output format: ")](reference)
print(output)