我使用的是python csv module,我有一个CSV,有3列,Item,Part,Category。在
我想创建一个dict,将所有类别组合起来,然后用项目:零件。在
例如:
512 SSD SATA,42-000153,Hardware
5M DisplayPort 1.2 Cable,42-000135,Cable
90W AC Adapter,42-000146,Adapter
4 port USB hub,42-000126,Adapter
我得到的结果是:
^{pr2}$这几乎让我想到了:
def build_dict(source_file):
projects = defaultdict(dict)
headers = ['Product', 'Part Number', 'Category']
with open(source_file, 'rb') as fp:
reader = csv.DictReader(fp, fieldnames=headers, dialect='excel',
skipinitialspace=True)
for rowdict in reader:
if None in rowdict:
del rowdict[None]
category = rowdict.pop("Category")
projects[category] = rowdict
return dict(projects)
source_file = 'test.csv'
我要找的结果是:
mydict = {
Hardware:{512 SSD SATA:42-000153},
Cable:{5M DisplayPort 1.2 Cable,42-000135},
Adapter:{90W AC Adapter:42-000146,4 port USB hub:42-000126}
}
请帮忙!在
我将利用Python的内置功能来实现:
输出:
^{pr2}$FWIW,您也可以这样做,这需要更多的代码行,但会使内部循环中发生的事情更具可读性。结果是一样的。注意它使用
csv.reader
而不是csv.DictReader
。在这可能有用。在
只需更改defaultdict以为每个项构建一个列表,您的代码将再次工作。在
输出:
^{pr2}$使用json pretty print(感谢martineau)
相关问题 更多 >
编程相关推荐