在Python中读取csv文件并创建一个字典

Question

我试图在python 27中读取一个csv文件来创建一个字典。 CSV文件看起来像 -

SI1440269，SI1320943，SI1321085 SI1440270，SI1320943，SI1321085，SI1320739 SI1440271，SI1320943
SI1440273，SI1321058，SI1320943，SI1320943

每行中的条目数不固定。 第一列条目应该是我的密钥。 我的代码 -

import csv
reader = csv.reader(open('test.csv'))

result = {}
for column in reader:
    key = column[0]
    if key in result:
        pass
    result[key] = column[1:]
print result

输出：

{'SI1440273'：['SI1321058'，'SI1320943'，'SI1320943']，''：[''，''，'']，'SI1440271'：['SI1320943'，''，'']，' SI1440270'：['SI1320943'，'SI1321085'，'SI1320739']，'SI1440269'：['SI1320943'，'SI1321085'，'']}

如何在输出中删除空值？ 另外，如何让输出中的键值与csv文件中的键值相同？

编辑：我想要每个'键'单行

Answer 1

您可以使用csv.DictReader ，如下所示：

import csv

result = {}
with open('test.csv') as csvfile:
    reader = csv.DictReader(csvfile, delimiter=" ", fieldnames=["id"], restkey="data")
    for row in reader:
        print row
        result[row["id"]] = row["data"]

print result

这将为您提供每行字典解决方案，因此您可以一次处理一行。 然后我还将它们全部附加到一个result字典中。

从这里你将得到以下输出：

{'data': ['SI1320943', 'SI1321085'], 'id': 'SI1440269'}
{'data': ['SI1320943', 'SI1321085', 'SI1320739', 'SI1440271', 'SI1320943'], 'id': 'SI1440270'}
{'data': ['SI1321058', 'SI1320943', 'SI1320943'], 'id': 'SI1440273'}
{'SI1440273': ['SI1321058', 'SI1320943', 'SI1320943'], 'SI1440270': ['SI1320943', 'SI1321085', 'SI1320739', 'SI1440271', 'SI1320943'], 'SI1440269': ['SI1320943', 'SI1321085']}

Answer 2

尝试这个

import csv
reader = csv.reader(open('test.csv'))

result = {row[0]:row[1:] for row in reader if row and row[0]}
print result

如果你想进一步消除值中的null，那么请执行下面的操作

import csv
reader = csv.reader(open('test.csv'))

result = {row[0]:[i for i in row[1:] if i] for row in reader if row and row[0]}
print result

保持进入顺序

from collections import OrderedDict
result = OrderedDict()
for row in reader:
   if row and row[0]:
      result[row[0]]=[i for i in row[1:] if i]

# print result
for key in result:
   print key,":" ,result[key]

Answer 3

如前所述，这不是CSV - 因此readline和split更合适，并使用OrderedDict来保持输入顺序：

import csv
from collections import OrderedDict
result = OrderedDict()
with open('test.csv') as f:
    for row  in f:
        row=row.strip().split()
        key = row[0]
        result[key] = row[1:]
print result