【发布时间】:2020-09-09 20:09:31
【问题描述】:
如果我有表数据 = [[1, 2, 0.5], [2, 5, 0.49]] 。我如何创建一个函数,在两行中找到第三列的最小元素,然后返回该数字所在的行?
例如。因为第三列中的最小数字是两行中的 0.49。我希望函数返回[2, 5, 0.49]
【问题讨论】:
如果我有表数据 = [[1, 2, 0.5], [2, 5, 0.49]] 。我如何创建一个函数,在两行中找到第三列的最小元素,然后返回该数字所在的行?
例如。因为第三列中的最小数字是两行中的 0.49。我希望函数返回[2, 5, 0.49]
【问题讨论】:
使用min() function 和key 比较列表的第三个元素。
min(data, key=lambda l: l[2])
【讨论】:
这样怎么样:
td = [[1, 2, 0.5], [2, 5, 0.49]]
col = 2 # third column
min_row = td[0] # initialze to something
for row in td:
if row[col] < min_row[col]:
min_row = row
print(min_row)
【讨论】: