通過用戶交互在最小值和最大值之間切換

Question

我有以下代碼，允許用戶在min和max過濾之間進行選擇。 但是， min適用於元素1但max適用於元素2 。 有沒有更好的方法來實現它？

from operator import itemgetter

data = [['A', '2', '4'], ['B', '2', '12'],
        ['C', '3', '88'], ['D', '4', '88']]
 
fn_max = False

if fn_max is True:
    mx = max(data, key=itemgetter(2))[2]
    mx_values = [d for d in data if d[2] == mx]
else:
    mx = min(data, key=itemgetter(1))[1]
    mx_values = [d for d in data if d[1] == mx]   
    
print(mx_values)

Answer 1

您可以將它們放入 function：

from operator import itemgetter

data = [['A', '2', '4'], ['B', '2', '12'],
        ['C', '3', '88'], ['D', '4', '88']]


def my_func(fn_max):
    func, i = (max, 2) if fn_max else (min, 1)
    mx = func(data, key=itemgetter(i))[i]
    return [d for d in data if d[i] == mx]


fn_max = False
print(my_func(fn_max))

Output：

[['A', '2', '4'], ['B', '2', '12']]

Answer 2

這可以說更好（雖然不清楚你需要它做什么，所以很難說它是否真的會更好）：

data = [
    ['A', 2, 4], 
    ['B', 2, 12],
    ['C', 3, 88], 
    ['D', 4, 88]
]

def fn_filter(fn, column, data):
    value = fn(rec[column] for rec in data)
    return list(filter(lambda rec: rec[column] == value, data))


print(fn_filter(min, 1, data))
print(fn_filter(max, 2, data))

結果：

[['A', 2, 4], ['B', 2, 12]]
[['C', 3, 88], ['D', 4, 88]]

fn_filter允許您將任何 function fn應用於數據集data的特定列column ，並將返回data中在同一列中具有相同值的所有記錄的列表。