如何编辑numpy数组?

Question

在python语言中，我的numpy数组是([12,1])，我想除以它

the value is 
([[1],
[2],
[3],
[4],
[5],
[6],
[7],
[8],
[9],
[10],
[11],
[12]])

，我想让

([[1],[2],[3],[4]],
[[5],[6],[7],[8]]
[[9],[10],[11],[12]])

所以我想把它分成三份(所以，batch_size是3)如何编写代码?

Answer 1

重塑为(3,4)是否足够?否则就会出现(3,4,1)

>>> arr.reshape(3,4)
array([[ 1,  2,  3,  4],
[ 5,  6,  7,  8],
[ 9, 10, 11, 12]])
>>> arr.reshape(3,4,1)
array([[[ 1],
[ 2],
[ 3],
[ 4]],
[[ 5],
[ 6],
[ 7],
[ 8]],
[[ 9],
[10],
[11],
[12]]])

Answer 2

original_array = [[1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12]]
array1 = original_array[:4]
array2 = original_array[4:8]
array3 = original_array[8:]
print(array1)
print(array2)
print(array3)