docAnalyze.py 文件源码

def getEdges(docTypes):
    import csv
    csv.field_size_limit(2147483647)
    for docType in docTypes:
        print(docType)
        with open("../output/edgelists/{}-edgelist.csv".format(docType.lower()), "r") as csvfile:
            datareader = csv.reader(csvfile)
            count = 0
            for row in datareader:
                if row[9].lower() in docTypes:
                    yield (row[0], row[2])
                    count += 1
                elif count < 2:
                    continue
                else:
                    return