python - 将 csv(不统一)拆分为 2 个列表

Question

我有一个 CSV 格式的数据

1,2,3,4,A
2,3,B
1,4,5,6,C
5,D

等等。我如何拆分它并将每个分配给 python 列表。 csv 中每行的长度是任意的。我需要将它们分成

   [[1,2,3,4],[2,3],[1,4,5,6],[5]]

其余的放入一个python列表中

['A','B','C','D']

完成此任务的有效方法是什么？下面的代码足以将 csv 转换为列表列表，但包含最后一个元素。我想要一个单独列表中的最后一个元素

with open('dataset.csv', 'r') as fp:
  reader = csvreader(fp)
  li = list(reader)

Answer 1

您可以遍历阅读器对象并将项目拆分/解包为除最后一个 和最后一个项目 之外的所有项目。

with open('dataset.csv', 'r') as fp:
  reader = csv.reader(fp)
  lst1, lst2 = [], []
  for row in reader:
      *others, last = row # Python 2: others, last = row[:-1], row[-1]
      lst1.append(others)
      lst2.append(last)