python - 读取 CSV，从字典中获取列数据类型

Question

我想将 CSV 文件读入 Pandas Dataframe。我有一个字符串列和一个 float 列。这是我的代码:

import pandas as pd
import decimal
D = decimal.Decimal

pd.read_csv(sourceFileName,
            usecols=["stringCol", "floatCol"],
            converters={"stringCol" : str,
                        "floatCol"  : lambda x: D(0) if x is "" else D(x)})

我想要做的是将列名称和类型存储在字典中，然后在 read_csv 语句中引用该字典的元素。

类似这样的事情:

colFormats = {"stringCol" : "str",
              "floatCol"  : "ModifiedDecimalFunction"}

有办法做到这一点吗？

非常感谢

Answer 1

您可以创建另一个字典，描述每种类型的含义:

formatConverters = {'str': str, 'floatCol': lambda x: D(0) if x is "" else D(x)}

现在使用

pd.read_csv(
    ....,
    converters={c: formatConverters[v] for (c, v) in colFormats.items()})