Search code examples
pythonpandasfilenames

Column with end of file name python

I have a code that merges all txt files from a directory into a dataframe

follow the code below

import pandas as pd
import os
import glob

diretorio = "F:\PROJETOS\LOTE45\ARQUIVOS\RISK\RISK_CUSTOM_FUND_N1" 
files = [] 

files = [pd.read_csv(file, delimiter='\t')
     for file in glob.glob(os.path.join(diretorio ,"*.txt"))]


df = pd.concat(files, ignore_index=True)
df

that gives result to this table

enter image description here

I needed to add a date column to this table, but I only have the date available at the end of the filename.

enter image description here

How can I get the date at the end of the filename and put it inside the dataframe.

I have no idea how to do this

Solution 
  • import pandas as pd
import os

diretorio = "F:/PROJETOS/LOTE45/ARQUIVOS/RISK/RISK_CUSTOM_FUND_N1/"
files = []
for filename in os.listdir(diretorio):
    if filename.endswith(".csv"):
        df = pd.read_csv(diretorio + filename, sep=";")
        df['Date'] = filename.split('.')[0].split("_")[-1]
        files.append(df)
df = pd.concat(files, ignore_index=True)
print(df)