Question

我来自这个问题Get information out of sub-lists in main list elegantly，提出了一个很好的解决方案（实际上是两个）。

我现在面临的问题是，给出的答案假定列表中的一定数量的元素（我从未说过元素的数量可能会改变，所以这是我的错），以生成最终列表。

这是代码的MWE：

a = [[0,1.1, 0.5, 99,99,0.7], [0,0.3, 1.4,99,99, 0.2], [1,0.6, 0.2,99,99, 1.], [1,1.1, 0.5,99,99, 0.3], [2,0.2, 1.1,99,99, 0.8], [2,1.1, 0.5,99,99, 1.], [3,1.2, 0.3,99,99, 0.6], [3,0.6, 0.4,99,99, 0.9], [4,0.6, 0.2,99,99, 0.5], [4,0.7, 0.2,99,99, 0.5]]

a_processed = []

from collections import defaultdict
nums = defaultdict(list)
for arr in a:
    key = tuple(arr[1:5]) # make the first two floats the key
    nums[key].append( arr[5] ) # append the third float for the given key

a_processed = [[k[0], k[1], k[2], k[3], round(sum(vals)/len(vals),2)] for k, vals in nums.items()]

返回：

a_processed = [[0.7, 0.2, 99, 99, 0.5], [0.3, 1.4, 99, 99, 0.2], [0.6, 0.2, 99, 99, 0.75], [0.6, 0.4, 99, 99, 0.9], [1.2, 0.3, 99, 99, 0.6], [0.2, 1.1, 99, 99, 0.8], [1.1, 0.5, 99, 99, 0.67]]

我想将显式命名元素k[0], k[1], k[2], k[3]的最后一行替换为更通用的内容。我试过简单地使用：

a_processed = [[k, round(sum(vals)/len(vals),2)] for k, vals in nums.items()]

但是将k存储为每个子列表中的元组：

a_processed = [[(0.7, 0.2, 99, 99), 0.5], [(0.3, 1.4, 99, 99), 0.2], [(0.6, 0.2, 99, 99), 0.75], [(0.6, 0.4, 99, 99), 0.9], [(1.2, 0.3, 99, 99), 0.6], [(0.2, 1.1, 99, 99), 0.8], [(1.1, 0.5, 99, 99), 0.67]]

那么如何将该行概括为任意数量的元素而不必明确地写下每一行？

Answer 1

使用list(k)将元组转换为列表，然后使用您需要包含的任何其他元素扩展该列表。

（您可以使用+运算符在python中连接列表。）

a_processed = [list(k) + [round(sum(vals)/len(vals),2)] for k, vals in nums.items()]

Answer 2

如果我的问题是正确的，那么以下就是您所寻求的。

a_processed = [list(k)[:4] + [round(sum(vals)/len(vals),2)] for k, vals in nums.items()]

+运算符按给定顺序加入2个列表。

将元组转换为元素列表

2 个答案: