inverseindex.py

def test():
    T = []
    T.append("it is what it is")
    T.append("what is it")
    T.append("it is a banana")

    ii = inverseIndex(T)

    for i in ii:
        print(i)

def makeInverseIndex(strlist):
    res = {}
    for (i, doc) in enumerate(strlist):
        for word in doc.split():
            if word in doc:
                if not word in res:
                    res[word] = set([i])
                else:
                    res[word].add(i)
    return res