python - 如何在python 2.7中自动将csv解析为int、string和list对象及其值

Question

我希望能够解析将在我们的软件组之外维护的 csv 文件。示例 csv 可能包含:

temperature,50,55,56
color,blue
count,10
dummy,line

代码将在文件中搜索名为“温度”的整数列表、一个名为“颜色”的字符串和一个名为“计数”的整数。然后它会分别分配值[50,55,56]，'blue'和10。

• csv 文件中的行顺序是任意的(例如，有时“颜色”可能位于“温度”之前)。
• 对象名称是预定义的，但 csv 文件之间的值会有所不同。
• csv 文件将始终包含 python 代码搜索的对象名称。
• csv 文件可能包含对 python 代码无用的额外数据行。
• 对象名称将始终位于 csv 的第一列中。

我见过其他搜索特定对象类型或预定义行位置中的对象的代码示例。但是，我需要一个更强大的解决方案。

任何帮助将不胜感激。如果您需要澄清，请告诉我，我会进行编辑。

Answer 1

您所需要的只是如下内容:

import csv

with open('file.csv', 'r') as f:
    reader = csv.reader(f)

    dummy = []
    temperature = []
    color = ''
    count = 0

    for row in reader:
        try:
            if row[0] == 'temperature':
                temperature = map(int, row[1:])
            elif row[0] == 'color':
                color = row[1]
            elif row[0] == 'count':
                count = int(row[1])
            else:
                dummy.append(row)

        # for robustness, let's catch the case where a row
        # may be malformed or does not contain more than 1 column
        except IndexError:
            dummy.append(row)