【问题标题】:Keep rows of a dataframe that are present in a list保留列表中存在的数据框的行
【发布时间】:2020-11-18 10:24:57
【问题描述】:

我有一个独特元素的列表,还有一个数据框,有很多列。我希望保留列表中也存在的数据框中的所有行。

例如,考虑拥有一个列表和一个数据框/csv 文件

列表:[a, c, e]

数据帧:

ColA    ColB
  a       1
  b       2
  c       3
  d       4
  e       5
  f       6

我希望将数据框修改为:

ColA    ColB
  a       1
  c       3
  e       5

我该怎么做?我希望保留和删除元素的 ColA 中的值都是唯一的。

【问题讨论】:

    标签: python python-3.x list dataframe


    【解决方案1】:

    使用Series.isin:

    In [2597]: df
    Out[2597]: 
      ColA  ColB
    0    a     1
    1    b     2
    2    c     3
    3    d     4
    4    e     5
    5    f     6
    
    In [2599]: l = ['a', 'c', 'e']
    
    In [2602]: df = df[df['ColA'].isin(l)]
    
    In [2603]: df
    Out[2603]: 
      ColA  ColB
    0    a     1
    2    c     3
    4    e     5
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2019-10-21
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-08-08
      • 2018-07-31
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多