使用-1为未知行重塑python数组并用None填充剩余空格

Question

我有一个长度未知的数组（例如让我们使用 11）。所以数组是[1,2,3,4,5,6,7,8,9,10,11]我想重塑该数组，以便他将拥有 5 列和尽可能多的行。我知道我可以使用reshape(-1,5) 这种方式根据数组长度创建行。但它给了我这个错误：ValueError: cannot reshape array of size 11 into shape (5)知道我该怎么做吗？期望的结果是：[[1,2,3,4,5],[6,7,8,9,10],[11,None,None,None,None]]我运行并收到此错误的代码是：import numpy as npa = np.array([1,2,3,4,5,6,7,8,9,10,11])    print(np.reshape(a, (-1,5)))

Answer 1

你可以在没有 numpy 的情况下做到这一点。

ar = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

n = 5

reshaped = [ar[i: i + n] + [None] * (i + n - len(ar)) for i in range(0, len(ar), n)]

您还可以将技巧与迭代器一起使用（块将在元组中）：

reshaped = list(zip(*[iter(ar + [None] * (n - len(ar) % n))] * n))

您可以zip_longest()从 itertools 申请不None自己添加值：

from itertools import zip_longest

reshaped = list(zip_longest(*[iter(ar)] * n))

Answer 2

In [135]: res = np.empty((3,5), object)                                                                      

In [136]: res                                                                                                

Out[136]: 

array([[None, None, None, None, None],

       [None, None, None, None, None],

       [None, None, None, None, None]], dtype=object)

In [137]: res.flat[:11] = np.arange(1,12)                                                                    

In [138]: res                                                                                                

Out[138]: 

array([[1, 2, 3, 4, 5],

       [6, 7, 8, 9, 10],

       [11, None, None, None, None]], dtype=object)