将 csv 数据与 RegEx、Python 匹配

Question

我有一个 csv 文件，其中包含以下行：

{"Books \u3009 Arts & Photography \u3009 Sculpture \u3009  Appreciation": 342444, "Books": 4244234234875}
{"Books \u3009 Science": 342864, "Books": 44242875}
{"Books \u3009 Politics": 342444, "Books": 4135875}

我需要读取 csv 文件并匹配 "Books": \d+，我只需要那个号码，并将其放入新的 csv 文件中

import csv
import json
import re
with open('input.csv', 'rb') as infile:
    with open('output.csv', 'wb',) as outfile:
        reader = csv.reader(infile)
        writer = csv.writer(outfile)
        for line_num, line in enumerate(reader):
            result = re.match(r'^"Books":(\d+)$', line[0])
            writer.writerow([result])

但是代码不能正常工作，基本上我收到空白单元格，这意味着我无法匹配所需的数据。如果我匹配.*，我会在 0x028B2800> 处收到“<_sre.sre_match>

Answer 1

这不是一个 CSV 文件，而是一堆 JSON 对象放在一起。它们看起来都是有效的，所以你可以解析它们并以这种方式提取数据：

import csv
import json

with open('input.csv', 'r') as infile, open('output.csv', 'w') as outfile:
    writer = csv.writer(outfile)
    for line in infile:
        num_books = json.loads(line.strip())['Books']
        writer.writerow([num_books])

在 python 3.4 中测试。