选择由元素列表组成的列的最后一个元素 pandas

Question

我有一个这样的熊猫数据框：

 df = pd.DataFrame([
    (142, [1,2]),
    (144, 3),
    (144, 1),
    (142, 1),
    (142, [1,3]),
    (144, [1,5]),
    (142, 2)
], columns=['ID', 'Column_1'])

如您所见，Column_1 是由单个元素和元素列表组成的列。

我的目标是创建一个新列，其中包含Column_1 是一个列表的最后一个元素，而单个值只是一个值，如下所示：

 df = pd.DataFrame([
    (142, [1,2] ,2),
    (144, 3 ,3),
    (144, 1 ,1),
    (142, 1 ,1),
    (142, [1,3] , 3),
    (144, [1,5] , 5),
    (142, 2)
], columns=['ID', 'Column_1' , 'new_column'])

谁能帮我解决这个问题？

Answer 1

让我们试试str 切片然后fillna，还要注意这里的最后一行应该是 2

df['new'] = df['Column_1'].str[-1].fillna(df.Column_1)
df
Out[118]: 
    ID Column_1  new_column  new
0  142   [1, 2]         2.0  2.0
1  144        3         3.0  3.0
2  144        1         1.0  1.0
3  142        1         1.0  1.0
4  142   [1, 3]         3.0  3.0
5  144   [1, 5]         5.0  5.0
6  142        2         NaN  2.0