使用python xlrd将xlsx文件的单个列写入csv

时间:2017-02-22 18:39:10

标签: python csv xlrd

我99%的路上......

def xl_to_csv(xl_file):
    wb = xlrd.open_workbook(xl_file)
    sh = wb.sheet_by_index(0)
    output = 'output.csv'
    op = open(output, 'wb')
    wr = csv.writer(op, quoting=csv.QUOTE_ALL)

    for rownum in range(sh.nrows):
        part_number = sh.cell(rownum,1)
        #wr.writerow(sh.row_values(rownum))  #writes entire row
        wr.writerow(part_number)
    op.close()

使用wr.writerow(sh.row_values(rownum))我可以将Excel文件中的整行写入CSV,但是有150列,我只想要其中一列。所以,我正在使用part_number = sh.cell(rownum,1)抓住我想要的一个列,但我似乎无法正确地将此变量写入CSV文件。

这是追溯:

Traceback (most recent call last):
  File "test.py", line 61, in <module>
    xl_to_csv(latest_file)
  File "test.py", line 32, in xl_to_csv
    wr.writerow(part_number)
_csv.Error: sequence expected

2 个答案:

答案 0 :(得分:2)

试试这个:

DoWork

参数必须是类似列表的对象。

答案 1 :(得分:1)

最快的解决方法是将partnum放入list(根据Abdou,您需要添加.value以获取单元格中的值):

for rownum in range(sh.nrows):
    part_number = sh.cell(rownum,1).value  # added '.value' to get value from cell
    wr.writerow([part_number]) # added brackets to give writerow the list it wants

更一般地说,您可以使用列表推导来获取所需的列:

cols = [1, 8, 110]
for rownum in range(sh.nrows):
    wr.writerow([sh.cell(rownum, colnum).value for colnum in cols])