我正在處理 cs50 的 pset6、DNA，我想讀取一個(gè)如下所示的 csv 文件：name,AGATC,AATG,TATCAlice,2,8,3Bob,4,1,5Charlie,3,2,5我想要?jiǎng)?chuàng)建的是一個(gè)嵌套字典，它看起來(lái)像這樣：data_dict = { "Alice" : { "AGATC" : 2, "AATG" : 8, "TATC" : 3 }, "Bob" : { "AGATC" : 4, "AATG" : 1, "TATC" : 5 }, "Charlie" : { "AGATC" : 3, "AATG" : 2, "TATC" : 5 }}所以我想用這個(gè)：with open(argv[1]) as data_file: for i in data_file:（或另一種變體）遍歷csv文件并append添加所有值到字典，以便我有一個(gè)以后可以訪問(wèn)的數(shù)據(jù)庫(kù)。
2 回答
慕尼黑8549860
TA貢獻(xiàn)1818條經(jīng)驗(yàn) 獲得超11個(gè)贊
你應(yīng)該使用 python 的csv.DictReader模塊
import csv
data_dict = {}
with open(argv[1]) as data_file:
reader = csv.DictReader(data_file)
for record in reader:
# `record` is a OrderedDict (type of dict) of column-name & value.
# Instead of creating the data pair as below:
# ```
# name = record["name"]
# data = {
# "AGATC": record["AGATC"],
# "AATG": record["AATG"],
# "TATC": record["TATC"],
# ...
# }
# data_dict[name] = data
# ```
# you can just delete the `name` column from `record`
name = record["name"]
del record["name"]
data_dict[name] = record
print(data_dict)
海綿寶寶撒
TA貢獻(xiàn)1809條經(jīng)驗(yàn) 獲得超8個(gè)贊
使用簡(jiǎn)單的文件讀取
with open(argv[1], 'r') as data_file:
line = next(data_file) # get the first line from file (i.e. header)
hdr = line.rstrip().split(',') # convert header string to comma delimited list
# ['name', 'AGATC', 'AATG', 'TATC']
data_dic = {}
for line in data_file:
line = line.rstrip().split(',')
# name and dictionary for current line
data_dic[line[0]] = {k:v for k, v in zip(hdr[1:], line[1:])}
print(data_dic)
輸出
{'Alice': {'AATG': '8', 'AGATC': '2', 'TATC': '3'},
'Bob': {'AATG': '1', 'AGATC': '4', 'TATC': '5'},
'Charlie': {'AATG': '2', 'AGATC': '3', 'TATC': '5'}}
添加回答
舉報(bào)
0/150
提交
取消