python - 字符串以特定文本开头

Question

我正在尝试读取 csv 文件，并且必须根据某些条件执行列操作。它完全忽略我的 if 条件并执行 else 语句。经过大量故障排除后，我无法纠正它，这很痛苦。

代码如下:
Tweet 是我的推文栏名称...

inf = pd.read_csv('string.csv')
for r in inf : 
    if "RT @" in inf.Tweet :   
        inf["Engagements"] = 0  
    else : 
        inf["Engagements"] = inf["Favorite_Count"] + inf["Retweet_Count"]

inf.to_csv('string2.csv', index=False)

Answer 1

在 pandas 中使用数组，所以需要 numpy.where带有 str.contains 创建的 bool 掩码以 ^ 作为 string 的开头或使用 str.startswith :

inf["Engagements"] = np.where(inf["Tweet"].str.contains('^RT @'), 
                              0, 
                              inf["Favorite_Count"] + inf["Retweet_Count"])

示例:

inf["Engagements"] = np.where(inf["Tweet"].str.contains('^RT @'), 
                              0, 
                              inf["Favorite_Count"] + inf["Retweet_Count"])

print (inf)
   Favorite_Count  Retweet_Count     Tweet  Engagements
0               1              2  RT @ ddd            0
1               4              0        dd            4
2               5              7  dds RT @           12

<小时/>

inf["Engagements"] = np.where(inf["Tweet"].str.startswith('RT @'), 
                              0, 
                              inf["Favorite_Count"] + inf["Retweet_Count"])

print (inf)
   Favorite_Count  Retweet_Count     Tweet  Engagements
0               1              2  RT @ ddd            0
1               4              0        dd            4
2               5              7  dds RT @           12