在 Pandas 中选择具有定义列的最小值和最大值的行

die*_*dro 0 python max min pandas

我有以下数据框:

A,B,C,D   
10,1,2,3
 1,4,7,3
10,5,2,3
40,7,9,3
9,9,9,9
Run Code Online (Sandbox Code Playgroud)

我想从前一个只有两行的数据帧开始创建另一个数据帧。这两行的选择基于“A”列中的最小值和最大值。我想得到:

A,B,C,D   
 1,4,7,3
40,7,9,3
Run Code Online (Sandbox Code Playgroud)

你认为我应该使用一种 index.min e index.max 然后只选择两行然后附加到一个新的数据帧中吗?你有同样的其他建议吗?

感谢您的任何帮助,最好

CHR*_*HRD 5

IIUC你可以简单地用一个子集数据帧OR的条件df.A.min()和df.A.max():

df = df[(df.A==df.A.min())|(df.A==df.A.max())]
Run Code Online (Sandbox Code Playgroud)
df
    A   B   C   D
1   1   4   7   3
3   40  7   9   3
Run Code Online (Sandbox Code Playgroud)