按条件过滤大熊猫

Posted 2021-05-05

我有一个数据框示例如下所示。第一行是标题。

a  b   c   d   e
x  1   10  4   asd
y  3   12  5   aqe
z  4   14  6   rty
t  6   12  4   abd
v  7   4   8   yul

我想通过过滤d列的最小值来查找列成员。我尝试对它进行排序并选择0.元素，但列可能有2个最小值，如示例中所示。

输出应为“x”和“t”

有什么建议？

答案

使用boolean indexing和loc选择列a并最后转换为list：

L = df.loc[df['d'] == df['d'].min(), 'a'].tolist()
print (L)
['x', 't']

详情：

print (df['d'] == df['d'].min())
0     True
1    False
2    False
3     True
4    False
Name: d, dtype: bool