gpt4 book ai didi

python - 从数据框中检索一定数量的行

转载 作者:行者123 更新时间:2023-12-04 00:15:42 25 4
gpt4 key购买 nike

我有一个数据框,

df = pd.DataFrame({"X1": ["A", "B", "A", "B", "B","C","C","C"],
"X2": ['FOO','BAR' ,'FOO1', 'BAR1', 'FOO2','BAR2','FOO3','BAR3']})

X1 X2
0 A FOO
1 B BAR
2 A FOO1
3 B BAR1
4 B FOO2
5 C BAR2
6 C FOO3
7 C BAR3

现在我正在计算给出 A:2、B:3、C:3 的值,并且我想根据 A 的计数提取行。这样,我可以有一个数据框,其中有 2 行A、B 2 行、C 2 行。

所以输出应该是,

    X1  X2
0 A FOO
2 A FOO1
1 B BAR
3 B BAR1
5 C BAR2
6 C FOO3

最佳答案

使用 GroupBy.head与计数 A 值按 sum 比较值按 Series.eq对于 ==,按列 X1 排序:

N = df['X1'].eq('A').sum()
df = df.sort_values('X1').groupby('X1').head(N)
print (df)
X1 X2
0 A FOO
2 A FOO1
1 B BAR
3 B BAR1
5 C BAR2
6 C FOO3

关于python - 从数据框中检索一定数量的行,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64134632/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com