data_converter.py 文件源码

def sparse_file_to_sparse_list (filename, verbose=True):
    ''' Converts a sparse data file to a sparse list, so that :
    sparse_list[i][j] = (a,b) means matrix[i][a]=b'''
    data_file = open(filename, "r")
    if verbose: print ("Reading {}...".format(filename))
    lines = data_file.readlines()
    if verbose: print ("Converting {} to correct array")
    data = [lines[i].split(' ') for i in range (len(lines))]
    del lines #djajetic 11.11.2015 questionable
    if verbose: print ("Converting {} to sparse list".format (filename))
    return [[tuple(map(int, data[i][j].rstrip().split(':'))) for j in range(len(data[i])) if data[i][j] != '\n'] for i in range (len(data))]