Изменение формы Numpy массив

Question

Я следующий массив NumPy,

[[date1,num1],[date2,num2],[date3,num3],[date4,num4]] 

Я хочу, чтобы разделить его следующим образом:

[ [ [date1,num1],[date2,num2] ] , [ [date3,num3],[date4,num4] ] ] 

Может ли кто-то предложить?

Answer 1

a = np.array([['date1','num1'],['date2','num2'],['date3','num3'],['date4','num4']]) 

a.reshape((2,2,2)) 
array([[['date1', 'num1'], 
    ['date2', 'num2']], 

    [['date3', 'num3'], 
    ['date4', 'num4']]]) 

Answer 2

Вы можете использовать метод Numpy в reshape и -1 аргумента для изменения формы массивов произвольной длины между двумя формами вы укажете. Как так:

import numpy as np 

#Generate an array of the form you specify of arbitrary length 
arraylen = 10 
a  = np.array([ ['date'+str(i),'num'+str(i)] for i in range(arraylen*2)]) 

#Reshape the array per your specifications 
a.reshape((-1,2,2)) 

дает

array([[['date0', 'num0'], 
     ['date1', 'num1']], 

     [['date2', 'num2'], 
     ['date3', 'num3']], 
     ...