使用python在csv中删除不需要的行

时间:2019-01-25 20:00:34

标签: python-3.x csv

我有一个包含名称,地址和电话号码的csv文件

但是在电话号码中,插入了一些地址,我想删除电话号码的那一行,请检查此屏幕截图,然后您就会知道问题所在:

Screenshot from csv

1 个答案:

答案 0 :(得分:0)

您可以根据需要修改以下代码:

import csv
import re
import sys

data = list(csv.reader(sys.stdin))

column_to_validate = 4
pattern_to_match = r"^\(\d{3}\) \d{3}-\d{4}"

data = list(filter(lambda line: re.match(pattern_to_match, line[column_to_validate]), data))

csv.writer(sys.stdout).writerows(data)