如何在Ubuntu上使用pyspark调用用户定义的函数

Question

我正在使用python 3.5＆amp;我的Ubuntu 16.0上有spark2.2.0 我有一个python函数connectDB来连接DB2。

def connectDB:
  import ibm_db
  conn = my Connection String
  sql = "select * from DB limit 10"
  stmt = ibm_db.exec_immediate(conn, sql)        
  while ibm_db.fetch_row(stmt) != False:    
      print ("COM_NAME : ",  ibm_db.result(stmt,3))
 return;

我想从我的pyspark调用此函数connectDB。你能告诉我怎么打电话吗？