我有一本字典,像这样:
a = {'date' : ['2012-03-09', '2012-01-12', '2012-11-11'],
'rate' : ['199', '900', '899'],
'country code' : ['1', '2', '44'],
'area code' : ['114', '11', '19'],
'product' : ['Mobile', 'Teddy', 'Handbag']}然后我使用zip函数来连接这些值:
data = [(a,b,c+d,e) for a,b,c,d,e in zip(*a.values())]输出:
data = [('2012-03-09', '199', '1114', 'Mobile'),
('2012-01-12', '900', '211', 'Teddy'),
('2012-11-11', '899', '4419', 'Handbag')]如果我想让函数自己搜索“国家代码”和“区域代码”,然后将它们合并,该怎么办?有什么建议吗?
发布于 2013-03-08 19:58:12
一个用于合并“列”的泛型方法,允许您指定预期的列和预先合并的列:
def merged_pivot(data, *output_names, **merged_columns):
input_names = []
column_map = {}
for col in output_names:
start = len(input_names)
input_names.extend(merged_columns.get(col, [col]))
column_map[col] = slice(start, len(input_names))
for row in zip(*(data[c] for c in input_names)):
yield tuple(''.join(row[column_map[c]]) for c in output_names)您可以使用以下命令调用它:
list(merged_pivot(a, 'date', 'rate', 'code', 'product', code=('country code', 'area code')))传入:
'date', 'rate', 'code', 'product')code=('country code', 'area code'),因此输出中的code是通过合并<代码>D11和area code).形成的
输出:
>>> list(merged_pivot(a, 'date', 'rate', 'code', 'product', code=('country code', 'area code')))
[('2012-03-09', '199', '1114', 'Mobile'), ('2012-01-12', '900', '211', 'Teddy'), ('2012-11-11', '899', '4419', 'Handbag')]或者,稍微重新格式化:
[('2012-03-09', '199', '1114', 'Mobile'),
('2012-01-12', '900', '211', 'Teddy'),
('2012-11-11', '899', '4419', 'Handbag')]如果只需要单独处理每一行,也可以循环遍历它的输出,而不是在merged_pivot()生成器上调用list():
columns = ('date', 'rate', 'code', 'product')
for row in merged_pivot(a, *columns, code=('country code', 'area code')):
# do something with `row`
print row发布于 2013-03-08 19:42:10
您必须自己定义键的顺序(否则a.values会以任意顺序返回它)。我把你原来的字典重命名为dd
[(a,b,c+d,e) for a,b,c,d,e in zip(*(dd[k] for k in ('date', 'rate', 'country code', 'area code', 'product')))]返回
[('2012-03-09', '199', '1114', 'Mobile'),
('2012-01-12', '900', '211', 'Teddy'),
('2012-11-11', '899', '4419', 'Handbag')]https://stackoverflow.com/questions/15293142
复制相似问题