使用列表理解读取csv文件

Question

我这样读取csv文件：

links = []
with open(join(input_path,'data.csv')) as csvfile:
    reader = csv.DictReader(csvfile)
    for record in (reader):
        if record['link']:
            links.append(record['link']) 

现在我在试这个

^{pr2}$

但是当我返回record而不是record['link']我得到了一个dict，如何才能得到与第一个使用列表理解相同的结果呢？在

Answer 1

您在附加record['link']，因此您应该在理解中包括：

links = [record['link'] for record in reader if record['link']]

Answer 2

只需使用与调用links.append()中相同的表达式，即record['link']：

links = [record['link'] for record in reader if record['link']]