python - 在 Python 中将 5 星评级量表转换为 'Positive' 和 'Negative' 量表很困难

Question

我有一个 csv，其中包含 1-5 范围内的整数。我正在尝试编写一个新的 csv，它采用原始 csv 中的任何 1 和 2 并返回“负”，然后所有 3、4 和 5 返回为“正”。这是我到目前为止所拥有的(仅供引用，我正在使用 Pandas 和 TextBlob):

ratings = pd.read_csv(r'/Users/jackcarlson/Downloads/amazon-cell-phones-reviews/only_rating.csv')

sentimentals = []
sentlist = [1,2]

for i in ratings.itertuples():
    if i in sentlist:
            neg = True
    if neg:
        sentimentals.append("negative")
    else:
        sentimentals.append("positive")
print(sentimentals)

with open('/Users/jackcarlson/Downloads/starreview.txt', 'w') as filehandle:
    for listitem in sentimentals:
        filehandle.write('%s\n' % listitem)

这是我的 csv 的样子:

我的 for 循环每行仅返回“负数”。我确信这是一个简单的错误(我对 Python 很陌生)，任何帮助都会很棒。谢谢!

Answer 1

这应该有效。

for i in ratings.itertuples():
    if i in sentlist:
        sentimentals.append("negative")
    else:
        sentimentals.append("positive")