from nltk.corpus import wordnet def get_synonyms(word): synonyms = [] for syn in wordnet.synsets(word): for lemma in syn.lemmas(): synonyms.append(lemma.name()) return synonyms def get_antonyms(word): antonyms = [] for syn in wordnet.synsets(word): for lemma in syn.lemmas(): if lemma.antonyms(): antonyms.append(lemma.antonyms()[0].name()) return antonyms word_dict = {} words = ["happy", "sad", "good", "bad"] for word in words: synonyms = get_synonyms(word) antonyms = get_antonyms(word) word_dict[word] = {"synonyms": synonyms, "antonyms": antonyms} print(word_dict)