创建类似对象的 csv DictReader

Question

在我的脚本中，我正在将 csv 文件读入字典。

csvreader = csv.DictReader(csvfile, delimiter=';', quotechar='"')

CSV 结构是：

"Name";"type"
"Name1";"type1"
"Name2";"type2"

然后我循环遍历csvreader 对象并将这些行用作创建请求的参数。如果响应代码是 404，我想创建类似的字典 err_obj_dict，其结构类似 csv 阅读器，其中包含请求失败的行：

err_obj_dict = {'Name': [], 'type': []}

for row in csvreader:
        if row['type'] == 'virtualResource':
            objType = 'virtual'
        else:
            objType = row['type']
....
....
if resp_cd == 400:
    err_obj_dict['Name'].append(row['Name'])
    err_obj_dict['type'].append(objType)

然后我想通过err_obj_dict 类似的方式循环通过csvreader 并创建请求：

 for row in err_obj_dict:
     resp_cd = ''
     if row['type'] == 'virtualResource':
         objType = 'virtual'
     else:
         objType = row['type']

但是脚本给了我：

if row['type'] == 'virtualResource':
TypeError: string indices must be integers, not str

似乎 DictReader 正在创建具有不同结构的不同类型的字典。有人可以帮我创建类似字典的 ReadDict 或如何遍历 err_obj_dict 以获得与 csvreader 相同的结果吗？

Answer 1

如果你打印 csvreader 的结果，这就是输出。

for row in csvreader:
    print(row)

{'type': 'type1', 'Name': 'Name1'}
{'type': 'type2', 'Name': 'Name2'}

因此，我建议使用字典数组而不是具有两个键 Name 和 Type 的字典。

err_obj_array = []
for row in csvreader:
    if row['type'] == 'virtualResource':
        objType = 'virtual'
    else:
        objType = row['type']
....
....
if resp_cd == 400:
  err_obj_dict = []
  err_obj_dict['Name'] = row['Name']
  err_obj_dict['type'] = objType
  err_obj_array.append(err_obj_dict)

当你想再次循环时，你可以这样做

for row in err_obj_array:
 resp_cd = ''
 if row['type'] == 'virtualResource':
     objType = 'virtual'
 else:
     objType = row['type']