Объединение нескольких файлов CSV с Python - PullRequest
2 голосов
/ 25 февраля 2020

Я пытаюсь объединить несколько файлов CSV в 1 файл CSV в сценарии python. Я хочу пропустить запись первых 5 строк каждого CSV-файла. Возникли проблемы, и я новичок в Python. Я пробовал несколько примеров, которые я нашел, но, похоже, проблемы с рабочим каталогом. Вот моя последняя попытка:

import pandas as pd
import csv
import glob
import os

path = '//server01/tmp/'
files_in_dir = [f for f in os.listdir(path) if f.endswith('csv')]
count = 0
for filenames in files_in_dir:
    df = pd.read_csv(filenames)
    if count < 6:
            count += 1
            continue
    df.to_csv('out.csv', mode='a')

Любая помощь будет оценена. Спасибо!

1 Ответ

3 голосов
/ 25 февраля 2020

Попробуйте:

import pandas as pd
import csv
import glob
import os

path = '//server01/tmp/'
files_in_dir = [os.path.join(path,f) for f in os.listdir(path) if f.endswith('csv')]
for filenames in files_in_dir:
    df = pd.read_csv(filenames, skiprows = 5)
    df.to_csv('out.csv', mode='a')

skiprows : количество пропускаемых строк

nrows : количество строк файла для чтения

...