使用langdetect计算熊猫数据框中的语言频率

时间:2019-01-14 21:21:29

标签: python pandas frequency

我想在推文数据集中找到不同语言的频率。我最终只想使用英语的推文,但也想了解其他语言的频率。

我已经使用langdetect检测到数据集中的推文的语言,现在我想计算每种语言的频率。这是我用于检测语言的代码:

from langdetect import detect    
import pandas as pd
data_path = "./output_1.csv"
df =  pd.read_csv(data_path, index_col=0)

for index, row in df.iterrows():
    print(detect(row['text']))
    if detect(row['text']) == "en":
        print(row['text'])

我想使用列表属性计数来计数频率:

using the list i = ['en','fr','es','it','cs','pt']
d = {x:i.count(x) for x in i}
print d

如何在使用langdetect获取的数据上使用count属性?

1 个答案:

答案 0 :(得分:1)

要创建包含您可以使用的语言的单独列:

df['language'] = df['text'].apply(lambda x: detect(x))

然后计算您可以执行的频率:

pd.DataFrame(df.groupby('language').text.count().sort_values(ascending=False))