用Python编辑电子表格（通过CSV文件传递）

Question

我有一个包含以下字段的电子表格：

id age smoker do sport
1   35  yes    rare
2   40  no     frequently
3   20  no     never 
4   ..  ..     ..

我想创建一个Python脚本来编辑通过csv文件转换传递的电子表格。在

“是”变为1，“否”变为0，“稀有”变为0，“频繁”变为1，“从不”变为2。在

我将电子表格保存为csv文件，使用分隔符“；”和引号'”。在

现在我写了这个代码：

^{pr2}$

但我不知道如何继续。。。。在

有人能告诉我如何使用python进行这些更改吗？在

使用Python列表或字典更好吗？在

谢谢大家！在

Answer 1

既然您已经了解了Python的csv library，那么对于输入csv的每一行，创建一个包含所需更改的新行，并将其写入一个新的csv文件，这应该很简单。在

请注意csv阅读器如何将每一行视为一个列表。接下来，查看csv writer的^{}方法；它接受一个python列表并将其作为csv行写入。你需要做的就是一次读一行，做你想要的修改，然后把它吐出来给作者。使用您的代码：

for row in reader: #for each row in the input
    outrow = list(row) # make a copy of the row. I'm not sure if you NEED to do this, but it doesn't hurt.

    if outrow[2] == "yes": #if the value in the 3rd column, "smoker", is "yes"
        outrow[2] = 1 #change it to 1
    elif outrow[2] == "no": #if it's "no"
        outrow[2] = 0 #change it to 0.

    #repeat this process for outrow[3] (meaning column #4, "do sport")

    output.writerow(outrow)

您可能注意到python调用第3列2和第4列3这是因为python从0开始计数（所以第一列是第0列）。您应该能够按照这个例子来进行所有需要的更改。在

完成后别忘了关闭文件！在