Question

我正在使用csv.DictWriter从一组字典输出csv文件。我使用以下函数：

def dictlist2file(dictrows, filename, fieldnames, delimiter='\t',
          lineterminator='\n'):
    out_f = open(filename, 'w')

    # Write out header
    header = delimiter.join(fieldnames) + lineterminator
    out_f.write(header)

    # Write out dictionary
    data = csv.DictWriter(out_f, fieldnames,
              delimiter=delimiter,
              lineterminator=lineterminator)
    data.writerows(dictrows)
    out_f.close()

其中dictrows是字典列表，字段名提供应序列化为文件的标题。

我的词典列表（dictrows）中的一些值是数字的 - 例如浮动，我想指定这些格式。例如，我可能希望使用“％。2f”而不是完全精度序列化浮点数。理想情况下，我想指定某种映射，说明如何格式化每种类型，例如

{float: "%.2f"}

表示如果你看到一个浮点数，请用％.2f格式化它。是否有捷径可寻？我不想将DictWriter或类似的任何复杂的子类化 - 这似乎是非常通用的功能。

如何做到这一点？

我能想到的唯一其他解决方案是：不要乱用DictWriter的格式，只需使用十进制包将浮点数的小数精度指定为％.2，这将导致序列化。不知道这是否是更好的解决方案？

非常感谢你的帮助。

Answer 1

class TypedWriter:
    """
    A CSV writer which will write rows to CSV file "f",
    which uses "fieldformats" to format fields.
    """

    def __init__(self, f, fieldnames, fieldformats, **kwds):
        self.writer = csv.DictWriter(f, fieldnames, **kwds)
        self.formats = fieldformats

    def writerow(self, row):
        self.writer.writerow(dict((k, self.formats[k] % v) 
                                  for k, v in row.iteritems()))

    def writerows(self, rows):
        for row in rows:
            self.writerow(row)

未经测试。

在Python中为csv.writer指定格式

1 个答案: