根据 if 条件在 csv 文件中写入新列答案

【问题标题】：Writing a new column in csv file based on if condition根据 if 条件在 csv 文件中写入新列
【发布时间】：2017-06-08 09:40:55
【问题描述】：

我想添加一个新列，该列根据推文是否为转推的条件添加两个现有列。我的 python 代码有以下 csv 输出：csv image

现在，我想实现如果推文是转推的条件，即；

   file1 = open(string.csv, 'rb')
reader = csv.reader(file1)
for r in rows :
   if Tweet[r]!="RT @***" :
        new_col=[col[3],col[4]]
        newcol.append(new_col)
    else :
        new_col=0       
file1.close()   


file2 = open(file.csv, 'wb')
writer = csv.writer(file2)
writer.writerows(newcol)
file2.close()

我在实现它时遇到了麻烦。请帮忙。

【问题讨论】：

标签： python twitter sentiment-analysis

【解决方案1】：

请尝试更改此语法 file2 = open(file.csv, 'wb') to file2 = open(file.csv, 'a')

谢谢

【讨论】：

【解决方案2】：

您可以使用startswith() 方法检查字符串是否以子字符串开头。

import csv
with open('tweets.csv') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        if row[0].startswith("RT @"):
            // It's a retweet
        else:
            // It's not a retweet

【讨论】：