python - 导入示例进行测试(Python)

Question

对于以下问题，我无法找到答案。
我有我的程序可以对一个文件中的图形执行各种操作。在我的类定义和函数之后，我有了在一对图上调用哪些函数的基本结构。
现在我想要不同的额外文件，每个文件都有一对图表，以显示我程序的不同方面。
如何将该文件放入我的主程序中，就好像额外文件中的代码在我的主程序中一样？
我知道，这是一个基本问题，但我以前从未处理过。

举例说明:

class foo:
    def __init__(stuff)

    def plot(self):
        ...

    more stuff

# I want the following in an extra file #
# Two equal graphs

A = nx.Graph([(0, 1), (1, 2), (2, 3), (3, 7), (3, 4), (4, 5), (5, 6)])
B = nx.Graph([(0, 1), (1, 2), (2, 3), (3, 7), (3, 4), (4, 5), (5, 6)])
# That's it

x = foo(A, B)
x.plot

如果我还可以通过命令行指定我要加载的文件，那就更好了。

Answer 1

您可以将您的变量/图形对放在它们自己的 python 文件中，然后将它们导入您的主文件。

例如

例子1.py

A = nx.Graph([...])
B = nx.Graph([...])

例子2.py

A = nx.Graph([...])
B = nx.Graph([...])

主要.py

import example1 as ex1
import example2 as ex2

class foo:
    ...

x = foo(ex1.A, ex2.B)
x.plot()

如果您真的希望用户在命令行指定要使用的图对，请查看argparse和 importlib

import argparse
import importlib
import foo

parser = argparse.ArgumentParser(description='Example app')
parser.add_argument('input', nargs=1)
args = parse.parse_args()

mod = importlib.import_module(args.input)
x = foo(mod.A, mod.B)
x.plot()