如何将Python元组转换为二维表？

Question

输入是一个一维的元组：

data = ('a','b','c'.....'z');

输出是一个有 n 列（比如说 9 列）的表格（二维的）

table = ?what code here?

那么

print table
 ( ('a','b','c'...), ('k','l','m','n'...), ....)

有什么最简单的方法来实现这个呢？

Answer 1

我很惊讶这个功能居然不在 itertools 这个模块里，但看起来确实没有。以下是我的做法：

def segment_list(lst, step):
    """
    Segment lst into a list of step-length lists
    """
    segments = []
    while True:
        if len(lst) <= step:
            output.append(lst)
            break
        output.append(lst[0:step])
        lst = lst[step+1:]
    return segments

Answer 2

这样做怎么样：

table = tuple(data[n:n+9] for n in xrange(0,len(data),9))

这个代码会返回一个元组，内容是从 data 中每次取9个项目。如果你想改变每行的数量，只需要把代码中的两个9改成你想要的数字就可以了。

Answer 3

如果你觉得下面的内容太啰嗦，这里有个简短版本：

n = 9
table = zip(*[iter(data)]*n)

假设你有一个列表：

>>> data = range(1,101)
>>> data
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18,
19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34,
35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50,
51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66,
67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82,
83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98,
99, 100]

为你的列表创建一个迭代器：

>>> data_iter = iter(data)

现在使用 zip 函数来分割这个列表。注意，这部分很有趣哦！

>>> table = zip(*[data_iter]*9)
>>> pprint.pprint(table)
[(1, 2, 3, 4, 5, 6, 7, 8, 9),
 (10, 11, 12, 13, 14, 15, 16, 17, 18),
 (19, 20, 21, 22, 23, 24, 25, 26, 27),
 (28, 29, 30, 31, 32, 33, 34, 35, 36),
 (37, 38, 39, 40, 41, 42, 43, 44, 45),
 (46, 47, 48, 49, 50, 51, 52, 53, 54),
 (55, 56, 57, 58, 59, 60, 61, 62, 63),
 (64, 65, 66, 67, 68, 69, 70, 71, 72),
 (73, 74, 75, 76, 77, 78, 79, 80, 81),
 (82, 83, 84, 85, 86, 87, 88, 89, 90),
 (91, 92, 93, 94, 95, 96, 97, 98, 99)]

表达式 `[data_iter]*9` 会生成以下列表：

[ data_iter, data_iter, data_iter, data_iter, data_iter, 
data_iter, data_iter, data_iter, data_iter ]

表达式 *[data_iter]*9 会把这个变成九个参数传给 zip 函数。这个 zip 函数会生成一个包含元组的列表：

返回一个元组列表，每个元组包含来自每个参数序列的第 i 个元素。返回的列表长度会被缩短到最短参数序列的长度。

因为我们使用的是迭代器，每次 zip 函数需要新值时，它都会获取序列中的下一个值。想象一下，如果我们只是用列表以同样的方式，会有什么不同：

>>> table = zip(*[data]*9)
>>> pprint.pprint(table)
[(1, 1, 1, 1, 1, 1, 1, 1, 1),
 (2, 2, 2, 2, 2, 2, 2, 2, 2),
 (3, 3, 3, 3, 3, 3, 3, 3, 3),
 ...
 (99, 99, 99, 99, 99, 99, 99, 99, 99),
 (100, 100, 100, 100, 100, 100, 100, 100, 100)]

更新：我的同事指出，如果你有 NumPy 库的话，你可以这样做：

>>> data = numpy.array(range(1,101))
>>> data.reshape([10,10])
array([[  1,   2,   3,   4,   5,   6,   7,   8,   9,  10],
   [ 11,  12,  13,  14,  15,  16,  17,  18,  19,  20],
   [ 21,  22,  23,  24,  25,  26,  27,  28,  29,  30],
   [ 31,  32,  33,  34,  35,  36,  37,  38,  39,  40],
   [ 41,  42,  43,  44,  45,  46,  47,  48,  49,  50],
   [ 51,  52,  53,  54,  55,  56,  57,  58,  59,  60],
   [ 61,  62,  63,  64,  65,  66,  67,  68,  69,  70],
   [ 71,  72,  73,  74,  75,  76,  77,  78,  79,  80],
   [ 81,  82,  83,  84,  85,  86,  87,  88,  89,  90],
   [ 91,  92,  93,  94,  95,  96,  97,  98,  99, 100]])