Question

我想添加一个新列，根据tweet是否为Retweet的条件添加两个现有列。我从我的python代码中跟随csv输出：csv image

现在，我想实现条件，如果tweet是转发，即;

   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()

我在实施它时遇到了麻烦。请帮助。

Answer 1

请尝试更改此语法file2 = open（file.csv，＆＃39; wb＆＃39;）要file2 = open（file.csv，＆＃39; a＆＃39;）

谢谢

Answer 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

根据if条件在csv文件中编写新列

2 个答案: