代码之家 › 专栏 › 技术社区 › ikel

在dataframe中搜索值并导出到新列

dataframe pandas

ikel · 技术社区 · 5 年前

我有一个大数据集(大约1百万行) 在这个数据集中,我想在一列(或多列)中找到一些值

例如,

数据框包含

   col1 col2 col3
-------------------
      a   b    c
      d   e    f
      g   h    i
      j   k    l
      m   n    o

我要寻找的是搜索每一行,如果给定值存在,那么在新的子列表中输出一个“是”。

有什么帮助吗?

谢谢

1 回复 | 直到 5 年前

Erfan 5 年前

场景1:搜索整个数据帧

我们可以利用 DataFrame.eq 具有 any 在列轴上,所以对于每一行。这意味着,如果 a 在任何一列中有一行,我们得到 True 以下内容:

df['indicator'] = df.eq('a').any(axis=1)

  col1 col2 col3  indicator
0    a    b    c       True
1    d    e    f      False
2    g    h    i      False
3    j    k    l      False
4    m    n    o      False

场景2:对于某些列:

如果我们使用 iloc ,以选择前两列

df['indicator'] = df.iloc[:, :2].eq('d').any(axis=1)

  col1 col2 col3  indicator
0    a    b    c      False
1    d    e    f       True
2    g    h    i      False
3    j    k    l      False
4    m    n    o      False

推荐文章

user1245262 · 筛选Pandas数据帧时出现问题

1 年前

Foroand · 熊猫数据帧中的词频计数耗时过长

1 年前

user14696236 · 如何为每个对应的列创建一行[重复]

2 年前

Shawn Hemelstrand · 为什么我的自定义errorbar函数不能在R中工作?

2 年前

Karim Abou El Naga · 将带字符串的DataFrame绘制到堆叠条形图中

2 年前

The Great · 拆分并存储数据帧,但名称基于特定列中的唯一值

2 年前

nickolakis · 基于R中的列名复制列

2 年前

opposity · 形成一个数据帧,该数据帧包含R中包含类别和子类别的列

2 年前

A. Handler · 有没有办法将数据帧的列与完整列名向量相匹配?

2 年前

JasonX · 运行减法计算

2 年前