'''
Data:2017-07-13
Auther;JXNU Kerwin
Description:使用Pandas拼接多个CSV文件到一个文件（即合并）
'''
import pandas as pd
import os
# 要拼接的文件夹及其完整路径，注意不要包含中文
Folder_Path = "/home/dengzeyi/coding/sequence_dos/cicids2017/input"
SaveFile_Path = "/home/dengzeyi/coding/sequence_dos/cicids2017/input"  # 拼接后要保存的文件路径
SaveFile_Name = "cicids-label.csv"  # 合并后要保存的文件名

# 修改当前工作目录
os.chdir(Folder_Path)
# 将该文件夹下的所有文件名存入一个列表
file_list = os.listdir()

# 排序
file_list.sort()

# 读取第一个CSV文件并包含表头
print(file_list[0])
df = pd.read_csv(Folder_Path + '/' + file_list[0])  # 编码默认UTF-8，若乱码自行更改

# 将读取的第一个CSV文件写入合并后的文件保存
df.to_csv(SaveFile_Path+'/' + SaveFile_Name, encoding="utf_8", index=False)

# 循环遍历列表中各个CSV文件名，并追加到合并后的文件
for i in range(1, len(file_list)):
    print(file_list[i])
    df = pd.read_csv(Folder_Path + '/' + file_list[i])
    df.to_csv(SaveFile_Path+'/' + SaveFile_Name, encoding="utf_8",
              index=False, header=False, mode='a+')