import pandas as pd
from snownlp import SnowNLP
def get_sentiment(comment):
values = SnowNLP(comment).sentiments
if values>0.6:
return "好评"
elif values<0.3:
return "差评"
else:
return "中评"
data = pd.read_csv("comment.csv",
sep=" ",
skiprows=1,
quotechar='"',
names=['product','comment']
)
data['sentiments'] = data['comment'].apply(get_sentiment)
data.to_csv("comment_tag.csv",index=False)