【发布时间】：2020-02-06 11:28:02
【问题描述】：
我有一个名为 shoes_2016 的文件名，如何创建一个名为 year 的数据框列并根据文件名提取年份。
【问题讨论】：
【发布时间】：2020-02-06 11:28:02
【问题描述】：
用split提取年份，转换为整数并赋值给DataFrame：
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)
【讨论】：
-
你为什么要去掉面具，为什么需要它？
-
@PrathikKini - 因为我首先认为 OP 需要过滤，然后再次阅读问题并且 OP 需要新列
-
如果我的文件名是'c:/users/a/desktop/items/shoes_2016.xlsx'，我该如何拆分
-
@3xtimpacts - 答案已编辑。