从根目录导入python包

Question

我具有以下目录结构：

main_work/
 data_processing/
  data_extraction.py
  data_selector.py

data_selector.py：

from data_extraction import connect_to_db

如果我开始从事main_work工作并想做

from data_processing import data_selector as ds

它抛出一个错误：

ModuleNotFoundError: No module named 'data_extraction'

因此，data_selector.py中的导入仅考虑当前文件夹（main_work），而不会对data_processing进行更深入的研究。

我需要能够通过main_work执行以下操作： 从data_processing导入data_selector作为ds

以及来自data_processing的以下内容： 导入data_selector

我如何实现这两个功能？

Answer 1

您可能想将文件夹添加到PYTHONPATH，这样python就会知道在哪里查找要导入的模块。

Answer 2

您需要以下结构

main_work/
  data_processing/
    __init__.py
    data_extraction.py
    data_selector.py

和data_selector.py

中

from .data_extraction import connect_to_db

.指定应与data_selector模块位于同一目录