Python 如何去掉csv文件的表头

最新推荐文章于 2024-04-22 01:20:27 发布

m0_66497803

最新推荐文章于 2024-04-22 01:20:27 发布

阅读量2k

点赞数

文章标签： python

本文链接：https://blog.csdn.net/m0_66497803/article/details/128439309

版权

在参考了其他人的代码以后，以下的代码可以直接使用，不再需要每一个文件的单独去表头。

import csv, os

os.makedirs('headerRemoved', exist_ok=True)

# Loop through every file in the current working directory.
for csvFilename in os.listdir('.'):
    if not csvFilename.endswith('.csv'):
        #i=i+1
        #print(i)
        continue # skip non-csv files
        
    i = 0
    print('Removing header from ' + csvFilename + '...')

    # Read the CSV file in (skipping first row).
    csvRows = []
    csvFileObj = open(csvFilename)
    readerObj = csv.reader(csvFileObj)
    csvData = list(readerObj)
    for row in csvData:
        i = i+1
        if i == 1:
            continue # skip first row
        csvRows.append(row)
        
    #print(csvRows)
    csvFileObj.close()

    with open(csvFilename, 'w', encoding='utf-8', newline='') as file_obj:
    # 创建对象
        writer = csv.writer(file_obj)
    # 遍历，将每一行的数据写入csv
        for p in csvRows:
            writer.writerow(p)