通過將另一列拆分兩次在Pandas DataFrame中創建新列

Question

我有以下DataFrame：

                          NAME
0  this_is_a_sample_2017-09-08.csv
1  this_is_a_sample_2017-09-09.csv
2  this_is_a_sample_2017-09-10.csv
3  this_is_a_sample_2017-09-11.csv
4  this_is_a_sample_2017-09-12.csv
5  this_is_a_sample_2017-09-13.csv

這是創建它的代碼：

rows = ['this_is_a_sample_2017-09-08','this_is_a_sample_2017-09-09','this_is_a_sample_2017-09-10','this_is_a_sample_2017-09-11','this_is_a_sample_2017-09-12','this_is_a_sample_2017-09-13']
df = pd.DataFrame(rows, columns = ['NAME'])

我想要一個新列，其中只包含名稱的日期部分。 結果數據框架如下所示：

                          NAME       DATE
0  this_is_a_sample_2017-09-08.csv   2017-09-08
1  this_is_a_sample_2017-09-09.csv   2017-09-09
2  this_is_a_sample_2017-09-10.csv   2017-09-10
3  this_is_a_sample_2017-09-11.csv   2017-09-11
4  this_is_a_sample_2017-09-12.csv   2017-09-12

我嘗試了以下方法：

df['DATE'] = df.NAME.str.split(".")[0].split('_')[-1]

我以為我可以將列拆分兩次並從列表中獲取日期，但是當我使用[0]對其進行索引時，它將選擇第一行（即列表）中的值。

Answer 1

使用.str訪問器選擇系列的索引，即

 df['Date'] =  df['NAME'].str.split(".").str[0].str.split('_').str[-1]

NAME        Date
0  this_is_a_sample_2017-09-08  2017-09-08
1  this_is_a_sample_2017-09-09  2017-09-09
2  this_is_a_sample_2017-09-10  2017-09-10
3  this_is_a_sample_2017-09-11  2017-09-11
4  this_is_a_sample_2017-09-12  2017-09-12
5  this_is_a_sample_2017-09-13  2017-09-13

Answer 2

或使用正則表達式

df['Date'] = df['NAME'].str.extract('.*?_(\d+-\d+-\d+).csv', expand = False)


    NAME                            Date
0   this_is_a_sample_2017-09-08.csv 2017-09-08
1   this_is_a_sample_2017-09-09.csv 2017-09-09
2   this_is_a_sample_2017-09-10.csv 2017-09-10
3   this_is_a_sample_2017-09-11.csv 2017-09-11
4   this_is_a_sample_2017-09-12.csv 2017-09-12
5   this_is_a_sample_2017-09-13.csv 2017-09-13

或按照@Evan的建議使用

df['Date'] = df['NAME'].str.extract('([\d-]+)', expand = False)

Answer 3

不太確定為什么要使用“ split（'。'）”，您真正需要的就是：

df['DATE'] = df['NAME'].str.split('_').str[-1]

                          NAME        DATE
0  this_is_a_sample_2017-09-08  2017-09-08
1  this_is_a_sample_2017-09-09  2017-09-09
2  this_is_a_sample_2017-09-10  2017-09-10
3  this_is_a_sample_2017-09-11  2017-09-11
4  this_is_a_sample_2017-09-12  2017-09-12
5  this_is_a_sample_2017-09-13  2017-09-13