ちょいめも

物理/Python/Cの雑記帳

csv連結

#フォルダ内のcsvファイルを名前順に連結
#2つ目以降のファイルはヘッダーを削除して連結
#第一引数:csvが入ったフォルダ

import sys
import glob
import pandas as pd

argvs = sys.argv #引数取得
argc = len(argvs) #引数の個数

if argc != 2:
    print('arg1 : csv folder path\n')
    input('Press any key to exit\n')
    sys.exit()

csv_list = glob.glob(argvs[1] + '/*.csv') #フォルダ内ファイル取得

if len(csv_list) != 0:

    list = []
    list0 = pd.read_csv(csv_list[0])
    for file in csv_list[1:]:
        df_tmp = pd.read_csv(file)
        df_tmp = df_tmp.drop(df_tmp.index[0])
        list.append(df_tmp)    
    df = pd.concat(list, sort=False)
    df = pd.concat([list0, df], sort=False)

    # csv出力
    df.to_csv('comb_csv.csv', index=False , encoding='utf_8')
    										
else:
	print('Error:no csv file')
	input('Press any key to exit\n')
	sys.exit()