我想编写一个函数,可以查看数据框中指定列的最大或最小值,然后返回包含最大或最小值的行的整个数据框,并将它们放在底部。
我已经让该函数只返回包含最大或最小值的行。
我已经让该函数只返回包含最大或最小值的行。
def findAggregate(df, transType, columnName=None):
if transType == 'max1Column':
return df[df[columnName] == df[columnName].max()]
elif transType == 'min1Column':
return df[df[columnName] == df[columnName].min()]
鉴于下面的数据框,我想检查col2的最小值
原始数据框:
col1 col2 col3
blue 2 dog
orange 18 cat
black 6 fish
预期输出:
col1 col2 col3
blue 2 dog
orange 18 cat
black 6 fish
blue 2 dog
实际输出:
col1 col2 col3
blue 2 dog
idxmax
和idxmin
返回的是索引,而不是整数索引。问题没有指定索引类型。因此,如果索引不是“rangeindex”,这将会失败。我认为像其他答案中所使用的loc
更加安全。 - Andy L.