python - 从带有文件名和键的文件创建字典

Question

当我测试这个时，我现在只获取列表中的日期，而不是列表中的位置、日期和标题。谁能帮我解决这个问题吗？最终结果应该是这样的。

{filename: [location, date, caption]}

这是文件。

images/skating.jpg,East York Arena,2014.11.03,Shea skating.,skating,Shea,boy
images/sunglasses.jpg,High Park,2013.02.03,Cool guy.,Shea,sunglasses,happy
images/skating2.jpg,East York Arena,2014.11.03,Shea skating 
again!,skating,Shea

def create_image_dict(open_csv_file):
'''(file) -> dict of {str: list of str}

The open csv file has the format:
filename,location,date,caption,keywords,keywords, ...
Return a dictionary with key filename and values [location, date, caption]
'''
d = {}
for line in open_csv_file:
    info = line.split(',')
    filename = info[0]
    location = info[1]
    date = info[2]
    caption = info[3]
    if location not in d:
        d[filename] = {location:{date: caption}}
return d

Answer 1

专业提示 1:

Do debug print statements to see what is being worked on i.e. print(info)

问题很可能是由于拆分导致列表中只有一个条目。

arr = "this string will not split to more than one element".split(",")

arr 现在的长度为 1，因此将引发 arr[1] 错误

编辑

再次查看您的程序，您似乎检查了错误的内容以插入到 dict d 中。您应该检查 if filename in d 而不是 if location in d。