python - 从 CSV 表中提取子字符串

Question

我正在尝试清理 csv 表中的数据，如下所示:

KATY PERRY@katyperry
1,084,149,282,038,820
Justin Bieber@justinbieber
10,527,300,631,674,900,000
Barack Obama@BarackObama
9,959,243,562,511,110,000

我只想提取“@”句柄，例如:

@katyperry
@justinbieber
@BarackObama

这是我编写的代码，但它所做的只是一遍又一遍地重复表格的第二行:

import csv
import re
with open('C:\\Users\\TK\\Steemit\\Scripts\\twitter.csv', 'rt',  encoding='UTF-8') as inp:
    read = csv.reader(inp)
    for row in read:
        for i in row:
            if i.isalpha():
                stringafterword = re.split('\\@\\',row)[-1]
        print(stringafterword)

Answer 1

如果你愿意使用re，你可以在一行中获取字符串列表:

import re

#content string added to make it a working example
content = """KATY PERRY@katyperry
1,084,149,282,038,820
Justin Bieber@justinbieber
10,527,300,631,674,900,000
Barack Obama@BarackObama
9,959,243,562,511,110,000"""

#solution using 're':
m = re.findall('@.*', content)
print(m)

#option without 're' but using string.find() based on your loop:
for row in content.split():
    pos_of_at = row.find('@')
    if pos_of_at > -1: #-1 indicates "substring not found"
        print(row[pos_of_at:])

您当然应该将content字符串替换为文件内容。