在 python 中拆分一个路径列

Question

Hi I have a column with path like this:

path_column = ['C:/Users/Desktop/sample\\1994-QTR1.tsv','C:/Users/Desktop/sample\\1995-QTR1.tsv']

I need to split and get just the file name.

Expected output:

[1994-QTR1,1995-QTR1]

Thanks

Answer 1

Use str.extract :

df['new'] = df['path'].str.extract(r'\\([^\\]*)\.\w+$', expand=False)

The equivalent with rsplit would be much less efficient:

df['new'] = df['path'].str.rsplit('\\', n=1).str[-1].str.rsplit('.', n=1).str[0]

Output:

                                    path        new
0  C:/Users/Desktop/sample\1994-QTR1.tsv  1994-QTR1
1  C:/Users/Desktop/sample\1995-QTR1.tsv  1995-QTR1

regex demo

Answer 2

Use this or you can use regex to match and take what you want.

path.split("\\")[-1].split(".")[0]

Output:

'1994-QTR1'

Edit

new_col=[]
for i in path_column:
    new_col.append(i.split("\\")[-1].split(".")[0])
print (new_col)

NOTE: If you need it in a list, you can append it to a new list from the loop.

Output:

['1994-QTR1', '1995-QTR1']

Answer 3

Similarly to the above, but you don't need to declare the separator.

import os

path = "C:/Users/Desktop/sample\\1994-QTR1.tsv"
name = path.split(os.path.sep)[-1]
print(name)

Answer 4

You might harness pathlib for this task following way

import pathlib
import pandas as pd
def get_stem(path):
    return pathlib.PureWindowsPath(path).stem
df = pd.DataFrame({'paths':['C:/Users/Desktop/sample\\1994-QTR1.tsv','C:/Users/Desktop/sample\\1994-QTR2.tsv','C:/Users/Desktop/sample\\1994-QTR3.tsv']})
df['names'] = df.paths.apply(get_stem)
print(df)

gives output

                                   paths      names
0  C:/Users/Desktop/sample\1994-QTR1.tsv  1994-QTR1
1  C:/Users/Desktop/sample\1994-QTR2.tsv  1994-QTR2
2  C:/Users/Desktop/sample\1994-QTR3.tsv  1994-QTR3