def locateWord(word, wordsArr): if word in wordsArr: return wordsArr.index(word) else: idxs = [wordsArr.index(w) for w in wordsArr if word in wordpunct_tokenize(w)] return idxs[0]