创建包含特定列中条目的列表

Question

我正在尝试创建一个列表，其中包含 csv 文件的第二列到最后一列中的条目。import csvwith open("file.csv") as csvfile:     reader = csv.reader(csvfile)     table = [row for row in reader]col_2_and_rest = [row[1:] for row in table]我目前正在列表中获取列表，这不是我想要的；我只想要一个列表。有办法吗？

Answer 1

这个怎么样：

import csv

with open("file.csv") as csvfile:

     reader = csv.reader(csvfile)

     table = [row for row in reader]

col_2_and_rest = sum([row[1:] for row in table], [])

Answer 2

您可以使用functools.reduce：

import csv

from functools import reduce

from operator import iconcat

with open("csv.csv") as csvfile:

     reader = csv.reader(csvfile)

     table = [row for row in reader]

col_2_and_rest = reduce(iconcat, [row[1:] for row in table])