如何编辑 numpy 数组？答案

【问题标题】：How to edit the numpy array?如何编辑 numpy 数组？
【发布时间】：2023-01-17 15:02:37
【问题描述】：

在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）如何制作代码？

【问题讨论】：

标签： python numpy pytorch

【解决方案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]]])

【讨论】：

【解决方案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)

【讨论】：