Python 在使用 Usecol 时添加列

Question

我希望这很容易完成，但我遇到了问题，因为我从文本文件中选择列，但也想将文件名的一部分添加为最后一列或第四列。当然，我收到了“预期轴”的错误。

下面是我的代码:

import pandas as pd
files = ('file_20181006.txt')
df = pd.read_table('file_20181006.txt', encoding='cp1252', sep='^', usecols=(0,1,70))
df.columns = ['Row Type','Account Number','Assigned Balance']
print(df[df['Row Type'] == 1])

我想添加一个名为“分配日期”的列，它解析我的文件名，仅获取“20181006”，仅此而已。

Answer 1

IIUC，尝试:

 df['Assign Date'] = files[-12:-4]

或者如果您想要日期时间数据类型:

df['Assign Date'] = pd.to_datetime(files[-12:-4])