comment puis-je ajouter zéro au début et à la fin de tout un tableau

J'ai un tableau comme celui-ci:

contact_map = array([[1., 1., 1.,..., 1., 1., 1.],
 [1., 1., 1.,..., 1., 1., 1.],
 [1., 1., 1.,..., 1., 1., 1.],
...,
 [0., 0., 0.,..., 0., 0., 0.],
 [0., 0., 0.,..., 0., 0., 0.],
 [0., 0., 0.,..., 0., 0., 0.]])

dont chaque élément est quelque chose comme ceci:

contact_map[19] = array([1., 1., 1., 1., 1., 1., 1., 1., 0., 1., 1., 0., 1., 1., 1., 1., 1.,
 1., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0.,
 0., 0., 0., 0., 0., 1., 0., 1., 1., 1., 1., 0., 0., 1., 1., 0., 0.,
 0., 0., 0., 0., 0., 0., 1., 1., 1., 1., 0., 0., 0., 0., 1., 1., 0.,
 1., 1., 0., 0., 0., 1., 1., 0., 1., 0., 0., 0., 1., 1., 1., 1., 1.,
 1., 0., 1., 0., 0., 0., 0., 1., 0., 0., 1., 0., 1., 1., 1.])

len(contact_map) = 224

len(contact_map[19]) =100

Je souhaite modifier tous les éléments de contact_map de manière à ajouter "0" au début et à la fin de chaque élément, par exemple en changeant contact_map[19] en :

contact_map[19] = array([0.,1., 1., 1., 1., 1., 1., 1., 1., 0., 1., 1., 0., 1., 1., 1., 1., 1.,
 1., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0.,
 0., 0., 0., 0., 0., 1., 0., 1., 1., 1., 1., 0., 0., 1., 1., 0., 0.,
 0., 0., 0., 0., 0., 0., 1., 1., 1., 1., 0., 0., 0., 0., 1., 1., 0.,
 1., 1., 0., 0., 0., 1., 1., 0., 1., 0., 0., 0., 1., 1., 1., 1., 1.,
 1., 0., 1., 0., 0., 0., 0., 1., 0., 0., 1., 0., 1., 1., 1., 0,.])

len(contact_map[19]) = 102

et ainsi de suite, pour contact_map[0], contact_map[1],....

Quelqu'un peut m'aider?

Solution du problème

Vous pouvez utiliser numpy.padcomme ci-dessous:

>>> import numpy as np
>>> a = np.array([[1., 1., 1., 1., 1., 1.],
  [1., 1., 0., 0., 1., 1.], 
  [0., 1., 0., 0., 1., 1.]])
>>> a.shape
(3, 6)
>>> out = np.pad(a, [(0,0), (1, 1)], 'constant', constant_values=0) 
# pad first dimension ^^^ ^^^
#  ^^^ pad second dimension 
#  (first num for set how many pad at first
#   second num for... at last.)  
>>> out.shape
(3,8)
>>> out
array([[0., 1., 1., 1., 1., 1., 1., 0.],
 [0., 1., 1., 0., 0., 1., 1., 0.],
 [0., 0., 1., 0., 0., 1., 1., 0.]])

Votre entrée :

contact_map = np.random.randint(0,2, (224, 100))
print(len(contact_map))
# 224
print(len(contact_map[19]))
# 100
contact_map = np.pad(contact_map, [(0,0), (1, 1)], 'constant', constant_values=0)
print(len(contact_map[19]))
# 102
print(len(contact_map[0]))
# 102