在numpy数组python中的每个元素后面添加元素
在numpy数组python中的每个元素后面添加元素
提问于 2020-02-22 14:51:34
我刚刚从numpy开始，并尝试创建一个函数，该函数接受一个数组(x)，将其转换为np.array，并返回一个numpy数组，在每个元素之后添加0,0,0,0。
它应该是这样的：
输入数组: 4,5,6
产出: 4,0,0,0,0,5,0,0,0,0,6,0,0,0,0，0，0
我尝试了以下几点：
import numpy as np
x = np.asarray([4,5,6])
y = np.array([])
for index, value in enumerate(x):
y = np.insert(x, index+1, [0,0,0,0])
print(y)返回：
[4 0 0 0 0 5 6]
[4 5 0 0 0 0 6]
[4 5 6 0 0 0 0]因此，基本上，我需要将输出组合成一个numpy数组，而不是三个列表。
有人知道怎么解决这个问题吗？
非常感谢！
回答 2
Stack Overflow用户
回答已采纳
发布于 2020-02-22 15:01:33
使用numpy .zeros函数！
import numpy as np
inputArray = [4,5,6]
newArray = np.zeros(5*len(inputArray),dtype=int)
newArray[::5] = inputArray实际上，将索引0、5和10的所有值“强制”为4、5和6。
所以_____0 0 0 0
变成4 0 0 0 5 0 0 0
>>> newArray
array([4, 0, 0, 0, 0, 5, 0, 0, 0, 0, 6, 0, 0, 0 ,0])Stack Overflow用户
发布于 2020-02-22 15:01:57
我还没有使用numpy来解决这个问题，但是这段代码似乎返回了所需的输出：
a = [4,5,6]
b = [0,0,0,0]
c = []
for x in a:
c = c + [x] + b
print(c)我希望这能帮到你!
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/60353370
复制相关文章
相似问题
腾讯云开发者
