python数组合并、输入三行数据,python/numpy一次合并4行子数组

作为一种纯粹的numpytonic方法,您可以通过创建一个从分块数到行数的范围(分块数为

step

ARG

range

. 然后使用

np.split()

要分割阵列:

In [24]: def chunk_array(arr, ch):

...: x = arr.shape[0]

...: return np.split(a, np.arange(ch, x, ch))

...:

...:

演示:

In [25]: chunk_array(a, 4)

Out[25]:

[array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]])]

In [26]: chunk_array(a, 3)

Out[26]:

[array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]]), array([[0, 0, 0, 0, 0, 0, 0],

[0, 0, 0, 0, 0, 0, 0]])]

如果希望将分块数组连接起来,可以使用@jpp的答案

np.concatenate()

和

map

或者在列表理解上稍有不同。

In [75]: def chunk_array(arr, ch):

...: x = arr.shape[0]

...: return [np.concatenate(subs) for subs in np.split(arr, np.arange(ch, x, ch))]