Question

假设我打开了一个文件来读取它的JSON数据：

import json
with open('filename', 'r') as infile:
    data = json.load(infile)

对于一个pickle文件，我使用：

import pickle
with open('filename', 'r') as infile:
    data = pickle.load(infile)

现在我希望能够将我的代码段用于两种格式，即：

import json
import pickle

def read_data(filename, preferred_serializer)
    with open(filename, 'r') as infile:
        data = preferred_serializer.load(infile)
    return data

并将其称为：

data1 = read_data('file.pickle', pickle)
data2 = read_data('file.json', json)

此似乎可行，因为json和pickle共享相同的API load(infile)。但我想知道它是否偶然起作用，或者这是否是定义的行为和合理的方法。

Answer 1

这很好。模块只是对象，就像Python中的其他所有内容一样。

这不是偶然的，这是设计的，Python确实尝试在命名中保持一致，因此反序列化数据的函数通常称为load。有关其他示例，请参阅marshal.load()。或yaml.load()。

如果你想让它更通用，你可以存储load函数引用：

def read_data(filename, deserialize)
    with open(filename, 'r') as infile:
        data = deserialize(infile)
    return data

data1 = read_data('file.pickle', pickle.load)
data2 = read_data('file.json', json.load)

毕竟，函数也只是对象，现在你使你的函数独立于可调用函数的名称。

是否可以将模块的引用分配给变量，然后使用变量而不是模块？

1 个答案: