向数据框添加值并导出

时间:2019-02-17 11:00:39

标签: python dataframe tokenize

我正在尝试将两个值添加为数据框中的列表 一个是句子,另一个是句子标记化后我得到的单词列表

现在,我已经完成了以下代码

from nltk.tokenize import word_tokenize
example = ['Mary had a little lamb' , 
        'Jack went up the hill' , 
        'Jill followed suit' ,    
        'i woke up suddenly' ,
       'it was a really bad dream...']


def hi():
    for i in example:
        #print (word_tokenize(i),i)
        a=[i,word_tokenize(i)]

        print(a) 

预期输出为

数据框具有两列,即原始句子和该句子的标记

示例

原始句子|令牌

我的名字是max |我的名字是最大

这是Windows |这是Windows

1 个答案:

答案 0 :(得分:1)

df['Original Sentence'] = a[0]  
df['Tokens'] = a[1]

或者我们可以完全跳过您的功能:

df['Original Sentence'] = example
df['Tokens'] = [word_tokenize(i) for i in example]

编辑:
由于它似乎没有开始的数据框。

import pandas as pd
df = pd.DataFrame.from_dict({'Original Sentence': example,
                   'Tokens': [word_tokenize(i) for i in example]})
print(df) #to see your dataframe 
df.to_csv('mydata.csv') #To output your dataframe into a csv file  

其他格式:

df.to_sql(etc...) #Refer to comment below  

要将sql直接输出到数据库,需要特定于数据库的设置。请参考此处的示例: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.to_sql.html