如何在Python中将CSV文件读入字典?

6

我希望能够从一个csv文件中读取数据(psc.csv),其包含以下两列:

cellname,scrambling
UER1100A,128
UER1100B,129
UER1100C,130
UER1300A,1
UER1300B,2
UER1300C,3
UER1400H,128

将整个文件放入一个字典中,使字典看起来像这样:

{'UER1100A': '128' , 'UER1100B': '129' , 'UER1100C': '130' ,  ...}

我尝试使用以下代码来使用csv模块,但是它返回了混合输出并在分离的字典中。有什么解决方案?
我的代码:

#!/usr/bin/python3
import csv
with open('psc.csv', newline='') as pscfile:
    reader = csv.DictReader(pscfile)
    for row in reader:
        print(row)
1个回答

8

只需将每一行添加到字典中:

import csv

results = {}
with open('psc.csv', newline='') as pscfile:
    reader = csv.DictReader(pscfile)
    for row in reader:
        results[row['cellname']] = row['scrambling']

与其使用DictReader,我会在这里使用一个普通的reader,并在跳过第一行后直接将结果提供给dict()调用。
import csv

with open('psc.csv',newline='') as pscfile:
    reader = csv.reader(pscfile)
    next(reader)
    results = dict(reader)  # pull in each row as a key-value pair

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接