python - 如何根据列中元素的数量从 Pandas 数据框中删除一行

Question

在下面的 pandas.DataFframe 中:

df = 
    alfa    beta   ceta
    a,b,c   c,d,e  g,e,h
    a,b     d,e,f  g,h,k
    j,k     c,k,l  f,k,n

如何删除 alfa 的列值超过 2 个元素的行？这可以使用长度函数来完成，我知道但没有找到具体的答案。

df = df[['alfa'].str.split(',').map(len) < 3]

Answer 1

您可以使用 pandas.DataFrame.apply()

依次对每一行进行测试

print(df[df['alfa'].apply(lambda x: len(x.split(',')) < 3)])

给予:

  alfa   beta   ceta
1  a,b  d,e,f  g,h,k
2  j,k  c,k,l  f,k,n