Gab*_*iel 1 python tuples list
我来自这个问题从主列表中的子列表中优雅地获取信息,提出了一个很好的解决方案(实际上是两个).
我现在面临的问题是,给出的答案假定列表中的一定数量的元素(我从未说过元素的数量可能会改变,所以这是我的错),以生成最终列表.
这是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()]
Run Code Online (Sandbox Code Playgroud)
返回:
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]]
Run Code Online (Sandbox Code Playgroud)
我想将显式命名元素的最后一行替换为k[0], k[1], k[2], k[3]更通用的.我试过简单地使用:
a_processed = [[k, round(sum(vals)/len(vals),2)] for k, vals in nums.items()]
Run Code Online (Sandbox Code Playgroud)
但是它存储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]]
Run Code Online (Sandbox Code Playgroud)
那么如何将该行概括为任意数量的元素而不必明确地写下每一行?
使用将元组转换为列表list(k),然后使用您需要包含的任何其他元素扩展该列表.
(您可以使用+运算符在python中连接列表.)
a_processed = [list(k) + [round(sum(vals)/len(vals),2)] for k, vals in nums.items()]
Run Code Online (Sandbox Code Playgroud)