使用python3导出数据时删除csv文件中的多余列

Question

我在python3中编写了一个函数，该函数合并同一目录中的某些文件并返回一个csv文件作为输出，但是csv文件的问题是我多了一个列开头没有header且该列的其他行是从0开始的数字。您知道我如何写csv文件而没有得到多余的列吗？

Answer 1

您可以用,分割，然后使用切片来删除第一个元素。

示例：

original = """col1,col2,col3
0,val01,val02,val03
1,val11,val12,val13
2,val21,val22,val23
"""

original_lines = original.splitlines()
result = original_lines[:1]  # copy header
for line in original_lines[1:]:
    result.append(','.join(line.split(',')[1:]))

print('\n'.join(result))

输出：

col1,col2,col3
val01,val02,val03
val11,val12,val13
val21,val22,val23