在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中，但我还是没能从蛋文件中导入我的自定义函数。请看下面的截图。

setup.py

蛋文件在Databricks上

当我尝试使用python命令时，

import function2

我得到一个错误信息，说没有找到这个模块。我感谢任何帮助和评论。注：init.py文件未被使用

Answer 1

经过搜索，我找到了以下解决方案。

1）创建库笔记本。例如--"Lib"，里面有任何函数类（没有可运行的代码）。

2) 用命令将Lib中的所有类& 函数导入到main中。

%run ".Lib" (工作原理是：从Lib导入*)

3) 之后，你可以从Main笔记本上调用任何在Lib中使用的函数类。