python - 使用 python 3x 查找 csv 文件

Question

我想查找下面的 csv 文件并从名为“datatype”的字段返回值，传递映射、转换和端口名作为查找端口。

Mapping    transformation   portname    datatype
m_TEST_1    EXP_test_1     field_1       nstring
m_TEST_1    EXP_test_1      field_2      date/time

基本上，我想执行(从 csv_file 选择数据类型，其中映射=？和转换=？和端口名=？)

目前，我正在循环遍历 csv 文件的每一行以获取数据类型。有没有简单且更好的方法来做到这一点。

下面是我当前使用的代码。

            lkp_file = csv.DictReader(open(lkpfile))
            for row in lkp_file:
                if mapping.get('NAME')==row['Mapping']:
                    if frominstance==row['transformation']:
                        if fromfield==row['portname']:
                            fromdatatype=row['datatype']
                            break

Answer 1

您可以采用的最佳方法是 csv.DictReader 及其某种转换。

(映射、转换、端口名) 是否唯一？

如果是这样，你可以做类似的事情:

import csv

d = {}
with open("path-to.csv", "r") as f:
    for row in csv.DictReader(f, delimiter=",")):
        d[(row['Mapping'], row['transformation'], row['portname'])] = row['datatype']

您必须交换分隔符，就像在我的示例中一样，我使用逗号，而您提供给我们的文本中没有它们。