python - 在 Databricks 笔记本中使用用 Python 编写的自定义函数

Question

我创建了一个包含两个不同函数的 python 文件 (.py)。
例如，

def function1():
    print("Hello World")

def funtion2(a, b):
    y=a+b
    if y>=5:
        print("Correct")
    else:
        print("Wrong")

现在，我想在 Databricks Notebook 中使用这两个函数。我所做的就是按照此databricks post中写的步骤进行操作。尽管我成功创建了一个egg文件，后来将其作为库导入到databricks中，但我还是无法从egg文件中导入我的自定义函数。请检查下面的屏幕截图:

setup.py

Databricks 上的鸡蛋文件

当我尝试 python 命令时:

import function2

我收到错误消息，提示找不到此模块。我感谢任何帮助和评论。
注意:未使用 init.py 文件

Answer 1

在网上搜索一番后，我找到了以下解决方案:

• 1. 创建图书馆笔记本。

  For example - Lib with any functions/classes there (no runnable code).

• 要将从 Lib 到 Main 的所有类和函数导入到 Main 中，请使用以下命令:

  %run "./Lib" 
  

  (这将像:from Lib import *)
• 之后，您可以从 Main 笔记本调用 Lib 中使用的任何函数/使用类。

这是online post与信息。