使用 pandas 获取新列中满足多个条件的列索引

Question

以以下数据框为例：

df = pd.DataFrame({'Sample':['X', 'Y', 'Z'], 'Base':[2, 10, 3], 'A':[0,5,100], 'C':[0,10,7]})

我想添加一个名为df["indices"]列索引的新列df["A"]和/或df["C"]只要它们满足两个条件：

1. 必须大于 5

2. df["A"]/df["Base"]或者df["C"]/df["Base"]必须大于或等于 1

生成的数据框将是：

df = pd.DataFrame({'Sample':['X', 'Y', 'Z'], 'Base':[2, 20, 3], 'A':[0,6,100], 'C':[0,10,7], 'indices': ['','C','A,C']})

我可以为我的第一个条件获取 True 或 False 值，df[['A','C']] > 5但我无法让它与我的条件 2 一起使用，该条件基于我的数据框中的另一列。在新专栏中获取 True 的索引又是另一回事了。我想象有 apply 和 get_loc 或 index 的东西，但无论我如何尝试，我都无法让它工作。

Answer 1

让我们创建一个满足两个给定条件的布尔掩码，然后使用DataFrame.dot该掩码来获取索引：m = df[['A', 'C']].gt(5) & df[['A', 'C']].div(df['Base'], axis=0).ge(1)df['indices'] = m.dot(m.columns + ',').str.rstrip(',')  Sample  Base    A   C indices0      X     2    0   0        1      Y    10    5  10       C2      Z     3  100   7     A,C

Answer 2

当满足任意数量的条件时，您可以将df.loc值分配回该列。一个简单的方法是拥有其中 3 个，每个都有您想要的条件。如果您愿意，您也可以将它们串联起来np.where以实现相同的目标。import pandas as pddf = pd.DataFrame({'Sample':['X', 'Y', 'Z'],&nbsp;&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;'Base':[2, 10, 3],&nbsp;&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;'A':[0,5,100],&nbsp;&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp;'C':[0,10,7]})df.loc[(df['A'] / df['Base'] >=1) & (df['C'] / df['Base'] >=1), 'indicies'] = 'A,C'df.loc[(df['A'] / df['Base'] >=1) & (df['C'] / df['Base'] <1), 'indicies'] = 'A'df.loc[(df['A'] / df['Base'] <1) & (df['C'] / df['Base'] >=1), 'indicies'] = 'C'输出&nbsp; &nbsp;Sample&nbsp; &nbsp;Base&nbsp; &nbsp; &nbsp; A&nbsp; &nbsp; &nbsp;C&nbsp; &nbsp;indicies0&nbsp; &nbsp; &nbsp; &nbsp;X&nbsp; &nbsp; &nbsp; 2&nbsp; &nbsp; &nbsp; 0&nbsp; &nbsp; &nbsp;0&nbsp; &nbsp; &nbsp; &nbsp;NaN1&nbsp; &nbsp; &nbsp; &nbsp;Y&nbsp; &nbsp; &nbsp;10&nbsp; &nbsp; &nbsp; 5&nbsp; &nbsp; 10&nbsp; &nbsp; &nbsp; &nbsp; &nbsp;C2&nbsp; &nbsp; &nbsp; &nbsp;Z&nbsp; &nbsp; &nbsp; 3&nbsp; &nbsp; 100&nbsp; &nbsp; &nbsp;7&nbsp; &nbsp; &nbsp; &nbsp;A,C