将列名标题插入 Python 列表

Question

我正在尝试将列标题插入 Python 列表。我使用的 csv 文件没有列名，数据用逗号分隔。我正在尝试编写这个程序，这样我就不必每次在 ArcGIS 中制作每周地图时都将列名插入到 csv 文件中。如果需要，我还可以将 csv 转换为字典。无论哪种方式都可以。到目前为止，我有这段代码，它产生了一个可用的数据集。我知道如何按列名引用字典中的列。但同样的方法似乎不适用于将列名插入列位置。

import os
os.chdir('C:\Users\####\Desktop')
fileData = open('SOUTH_06_01_2015_SECFILE.CSV','rb')

def dataHeader():
    headerLine = fileData.readline()
    valueList = headerLine.split(',')
    yield valueList

def dataBody():
    for line in fileData.readlines():
        segmentLine = line.split(',')
        yield segmentLine[0:6]

def dataPrinter():
    gen1 = dataHeader()
    gen2 = dataBody()
    print'\n'
    for row in gen2:
        print row
dataPrinter()

Answer 1

假设用户插入了列名，而 csv 文件没有列名，这应该可以完成。

def dataHeader(columns_no):
    columns = []
    for i in range(columns_no):
        text = raw_input("Enter column #%d name> " % i)
        columns.append(text)
    return columns


def dataBody(columns_no):
    segmentLines = []
    for line in fileData.readlines():
        segmentLine = line.split(',')
        segmentLines.append(segmentLine[:columns_no])
    return segmentLines


def dataPrinter():
    columns_no = 6
    gen1 = dataHeader(columns_no)
    gen2 = dataBody(columns_no)
    print ','.join(gen1)
    for row in gen2:
        print ','.join(row)


dataPrinter()