如果我运行以下内容:
import numpy as np
a = np.arange(9)
a = a.reshape((3,3))
我会得到这个:
a = [[0 1 2]
[3 4 5]
[6 7 8]]
如果我像这样创建一个更大的数组:
b = np.zeros((5,5))
b = [[ 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.]]
如何有效地将a
复制到b
以获取此类数组?
# border of 0 surrounding a to be filled in with other data later
b = [[ 0. 0. 0. 0. 0.]
[ 0. 1. 2. 3. 0.]
[ 0. 4. 5. 6. 0.]
[ 0. 7. 8. 9. 0.]
[ 0. 0. 0. 0. 0.]]
我正在寻找numpy
内置的函数(如果它存在)。
答案 0 :(得分:14)
您可以指定b[1:4, 1:4]
来表示该部分:
>>> import numpy as np
>>> a = np.arange(9)
>>> a = a.reshape((3, 3))
>>> b = np.zeros((5, 5))
>>> b[1:4, 1:4] = a
>>> b
array([[ 0., 0., 0., 0., 0.],
[ 0., 0., 1., 2., 0.],
[ 0., 3., 4., 5., 0.],
[ 0., 6., 7., 8., 0.],
[ 0., 0., 0., 0., 0.]])
>>> b[1:4,1:4] = a + 1 # If you really meant `[1, 2, ..., 9]`
>>> b
array([[ 0., 0., 0., 0., 0.],
[ 0., 1., 2., 3., 0.],
[ 0., 4., 5., 6., 0.],
[ 0., 7., 8., 9., 0.],
[ 0., 0., 0., 0., 0.]])
答案 1 :(得分:3)
作为替代方案,如果您想要一个非零的垫片值,您可以使用此选项
>>> a = np.arange(9.).reshape(3,3)
>>> np.pad(a, 1, 'constant', constant_values=0)
array([[ 0., 0., 0., 0., 0.],
[ 0., 0., 1., 2., 0.],
[ 0., 3., 4., 5., 0.],
[ 0., 6., 7., 8., 0.],
[ 0., 0., 0., 0., 0.]])
>>> np.pad(a, 1, 'constant', constant_values=5)
array([[ 5., 5., 5., 5., 5.],
[ 5., 0., 1., 2., 5.],
[ 5., 3., 4., 5., 5.],
[ 5., 6., 7., 8., 5.],
[ 5., 5., 5., 5., 5.]])