请告诉我,如何使用锯齿形将列表转换为矩阵?相反,有代码以锯齿形方式将矩阵转换为列表:
def zigzag(a):
m, n = a.shape
solution = [[] for i in range(m + n - 1)]
for i in range(m):
for j in range(n):
sum = i + j
value = a[i][j]
index = (i, j)
if sum % 2 == 0:
solution[sum].insert(0, (value, index))
else:
solution[sum].append((value, index))
output_vector = []
for i in solution:
for j in i:
output_vector.append(j[0])
return output_vector


