Question

我为这个练习提出了两个不同的功能。一个使用csv类编写器，另一个使用DictWriter类。两者都通过了列出的预期结果，但它们未通过私人测试用例？

import csv
def csvWriter(filename, records): 
    header = []
    for i in records:
        if len(i) < 1:
          records.remove(i)
    for i in records:
        for v in i:
          if v not in header:
            header.append(v)
    for i in records:
        if len(i) == 0:
            return '0 records processed.'
    test=open(filename,'w') 
    wr = csv.writer(test,header,lineterminator='\n')
    wr.writerow(header)
    for i in records:
        wr.writerow(i.values())
    test.close()
    return '%d records processed.' % len(records)

csvWriter（ '文件名'，[{ 'A'：1， 'B'：2}，{ '一个'：3， 'B'：4}]

repr（open（'filename'）。read（））---＆gt; 'a，b \ n1,2，\ na，b \ n3,4'检查

'传递了2条记录。' ---＆GT; “2条记录通过了。” 检查

私人测试案例---＆gt;失败？

Answer 1

这个函数在私有测试用例上失败的原因是，可以传递第二个参数，并且值将被写入而不进行排序。

import csv
def csvWriter(filename, records): 
  header = []
  for i in records:
    if len(i) < 1:
      records.remove(i)
  for i in records:
    for v in i:
      if v not in header:
        header.append(v)
  for i in records:
    if len(i) == 0:
      return '0 records processed.'
  test=open(filename,'w') 
  dict_wr = csv.DictWriter(test,header,lineterminator='\n')
  dict_wr.writerow(dict(zip(header,header)))
  for i in records:     
  # Adding in the **sorted** built-in fixed it
    dict_wr.writerow(dict(zip(header,sorted(i.values()))))  
  test.close()
  return '%d records processed.' % len(records)

还是非常粗糙，我们也不想使用csv模块。我要回去改进这个并试一试没有模块。

有什么建议吗？

Answer 2

导入csv

def csvWriter（文件名，记录）：

header = []

f = []
with open(filename, 'w')as new_file:

    csv_writer = csv.writer(new_file,header,lineterminator='\n')

    for line in records:
        if line not in header:
            header.append(sorted(line))
            header.append(sorted(line.values()))
            for x in header:
                if x not in f:
                    f.append(x)
                    csv_writer.writerow(x)
return '%s records processed.' % len(records)

Pyschools Topic 13 Q 9 csv.writer

2 个答案: