Python中如何调用自己写的函数

例如我现在写了一个KNN的函数，如下：

import numpy as np

def createDataSet():
    group=np.array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
    labels=['A','A','B','B']
    return group,labels

在另一个.py文件中想引用的话

首先import kNN

引用函数kNN.createDataSet()