列表的数据框：获取长度大于1的值

Question

假设我有一个这样的数据框：

df = pd.DataFrame(data=np.array([[1],[1,2],[1,2,3],[2]]), columns=['col'])

我最终需要到达以下子列表：

    col
0   [1, 2]
1   [1, 2, 3] 

这意味着我需要“ 获取长度大于1的行”或“ 删除长度小于等于1的行”。我该怎么办？

Answer 1

您可以使用.loc进行过滤，并在len列的'col' gth上设置一个条件，然后在结果数据帧上使用reset_index来忽略索引旧的数据框：

print(df.loc[df.col.str.len() > 1].reset_index(drop=True))

输出：

         col
0     [1, 2]
1  [1, 2, 3]