将列中的所有唯一词放入新数据集中

Question

从专栏中获取唯一词并将其放在新列中

我尝试了以下代码，但是没有用：

query=list(train['doc_text'].str.split(' ', expand=True).stack().unique())

以下是数据示例：

Train
Row             Doc_text                 Count
0             this is a book               4
1             my taylor is rich            4 
2             apple a day                  3

以下是预期输出的示例：

Dfnew
Row         Uniquewords
0            this
1            is
2            a
3            book
4            my 
5            taylor
6            rich
7            apple
8            day    

我想获取列表中的单词，然后能够将此列表另存为新数据集。

Answer 1

您也可以这样做：

unique_list = []
for i in df['Uniquewords']:
    [unique_list.append(word) for word in i.split() if word not in unique_list]

您可以使用此unique_list，也可以将此列表写入数据框。

df_new = pd.DataFrame(unique_list, columns=['Unique_words'])

Answer 2

IIUC，您需要以下类似内容：

df_new=pd.DataFrame(train['doc_text'].str.split(' ', expand=True).stack().unique(),\
                columns=['Uniquewords']).reset_index().rename(columns={'index':'Row'})
print(df_new)

   Row Uniquewords
0    0        this
1    1          is
2    2           a
3    3        book
4    4          my
5    5      taylor
6    6        rich
7    7       apple
8    8         day