对压缩矩阵元素重新排序

Question

我在一个可变大小的电子表格中有两个匹配的大小矩阵，一个带有数据项，另一个为处理启用了标记项。拉链后的项目，他们不是在一个方便的顺序处理。项目是如何压缩的，所以我有项目和启用一起？你知道吗

期望只zipper = zip(output, doit)，但是失败得很严重，如输出中所示。你知道吗

预期输出和压缩矩阵如下所示。你知道吗

from __future__ import print_function
from openpyxl import Workbook
wb = Workbook()
ws = wb.active

rows = [
    ['Number', 'Batch 1', 'Batch 2'],
    [2, 40, 30],
    [3, 40, 25],
    [4, 50, 30],
    [5, 30, 10],
]
enabled = [
    [0, 0, 0],
    [0, 1, 0],
    [1, 1, 0],
    [0, 0, 1],
    [0, 1, 0],
]

for row in rows:
    ws.append(row)
output = []
for i, row in enumerate(ws['B1:C5']):
    output.append([])
    for cell in row:
        output[i].append(cell.value)

for row in enabled:
    ws.append(row)
doit = []
for i, row in enumerate(ws['A6:B10']):
    doit.append([])
    for cell in row:
        doit[i].append(cell.value)

zipper = zip(output, doit)
print(zipper)

for i in range(len(output[0])):
    print(">>Do column")
    for j in range(len(output)):
        if doit[j][i]:
            print(output[j][i])

输出

[([u'Batch 1', u'Batch 2'], [0, 0]), ([40, 30], [0, 1]), ([40, 25], [1, 1]), ([50, 30], [0, 0]), ([30, 10], [0, 1])]
>>Do column
40
>>Do column
30
25
10

我希望拉链看起来像：

[
[(0, u'Batch 1'), (0, u'Batch 2')]
[(0, 40), (1, 30)]
[(1, 40), (1, 25)]
[(0, 50), (0, 30)]
[(0, 30), (1, 10)]
]

在以下情况下不成功：

# Flatten
zipper = zip(sum(output, []), sum(doit, []))
# Reassemble array
y = zip(*[iter(zipper)]*2)
print(list(y))

Answer 1

这应该管用

zipper = [[(a, c), (b, d)] for [a, b], [c, d] in zip(output, doit)]
inverse_zipper = [[(c, a), (d, b)] for [a, b], [c, d] in zip(output, doit)]

取代你的老本行

zipper = zip(output, doit)

---

标准类型

[]表示list和()a tuple。python类型的文档是here。主要区别在于元组是不可变的。在这里我只是尊重你想要的产出

---

列表理解

zipper = [[(a, c), (b, d)] for [a, b], [c, d] in zip(output, doit)]

相当于

zipper = []
for [a, b], [c, d] in zip(output, doit):
    zipper.append([(a, c), (b, d)])

---

打开包装

开箱是一项很快的任务。a, b = [4,7]等价于

some_list = [4,7]
a = some_list[0]
b = some_list[1]

都将4分配给a，7分配给b

你知道zip(output, doit)的输出是[([u'Batch 1', u'Batch 2'], [0, 0]), ([40, 30], [0, 1]), ([40, 25], [1, 1]), ([50, 30], [0, 0]), ([30, 10], [0, 1])] 所以如果你做for row in zip(output, doit):，row将以([40, 30], [0, 1])的形式出现，它可以被解压成[a, b], [c, d] 您可以直接在for语句中进行赋值，该语句给出for [a, b], [c, d] in zip(output, doit)