1

我来自这个问题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]]

那么我怎样才能将该行推广到任意数量的元素，而不必明确写下每个元素呢？

4

2 回答 2

6

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

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

a_processed = [list(k) + [round(sum(vals)/len(vals),2)] for k, vals in nums.items()]
于 2013-10-24T22:02:07.117 回答
2

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

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

+ 运算符按给定顺序连接 2 个列表。

于 2013-10-24T22:07:20.640 回答