读取文件名并用i创建列

Question

我有一个名为shoes\u 2016的文件名，如何创建一个名为year的dataframe列，并根据文件名提取年份。你知道吗

Answer 1

用split提取年份，转换为整数并分配给数据帧：

filename = 'shoes_2016.csv'

df = pd.read_csv(filename)
df['year'] = int(filename.split('.')[0].split('_')[1])

print (df)

或：

filename = 'shoes_2016.csv'

df = pd.read_csv(filename).assign(year = int(filename.split('.')[0].split('_')[1]))
print (df)

编辑：

import os

filename = 'c:/users/a/desktop/items/shoes_2016.xlsx'
y = int(os.path.basename(filename).split('.')[0].split('_')[1])
print (y)
2016

或：

y = int(filename.split('/')[-1].split('.')[0].split('_')[1])
print (y)
2016

df = pd.read_csv(filename).assign(year = y)