可以使用Pandas库的groupby()函数加上apply()函数来实现。在apply()函数内通过lambda表达式使每个分组返回整行数据。具体代码如下:
import pandas as pd
# 创建DataFrame
df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]})
# 按'A'列分组并选择整行数据
df.groupby(['A']).apply(lambda x: x)
该代码将按照‘A’列的值进行分组,然后对每个分组应用lambda表达式,以返回整个数据行。最终的输出结果将是原始DataFrame的完整副本,其中每个分组的行被保留。
上一篇:按单独列隐藏空行
下一篇:按单个日期获取中位数