io.py 文件源码

def txt2jd(f):
    """
    Dump the np.ndarray using joblib.dump for fast access.
    """
    data = []
    for line in open(f):
        line = line.split("\n")[0].split("\t")
        data.append(map(int, line))
    data = np.array(data)
    joblib.dump(data, f.replace(".txt", ".jd"))
    os.remove(f)
    return f.replace(".txt", ".jd")