在做科学计算或者模拟仿真的时候,相信不少小伙伴会遇到这样的问题,比如,我们有一个一维数组如下所示:
array = [1, 2, 3, 4, 5]
此时,我们想要将其沿着 y 轴进行重复性堆叠,比如,这里我们设定 3 次, 从而我们可以得到下面的数组。
[[1. 2. 3. 4. 5.] [1. 2. 3. 4. 5.] [1. 2. 3. 4. 5.]]
那么我们该怎么办呢?
import numpy as np
array = np.array([1, 2, 3, 4, 5]) # 原始数组
repeat_time = 3 # 沿着y轴堆叠的次数
array_final = np.ones([repeat_time, len(array)])
for i in range(repeat_time):
array_final[i, :] = array
print(array_final)
"""
result:
[[1. 2. 3. 4. 5.]
[1. 2. 3. 4. 5.]
[1. 2. 3. 4. 5.]]
"""显然,上述方法比较麻烦,为了简化,我们可以世界使用 np.repeat() 函数实现这个功能。
import numpy as np array = np.array([1, 2, 3, 4, 5]) # 原始数组 repeat_time = 3 # 沿着y轴堆叠的次数 array_final = np.repeat(array.reshape(1, -1), axis=0, repeats=repeat_time) print(array_final) """ result: [[1 2 3 4 5] [1 2 3 4 5] [1 2 3 4 5]] """
关于 np.repeat() 函数的详细用法可以参考这篇------np.repeat()函数。
立即学习“Python免费学习笔记(深入)”;
当然了,对于这种情况,最简单的方法就是使用 np.meshgrid() 函数来处理。
import numpy as np array = np.array([1, 2, 3, 4, 5]) # 原始数组 repeat_time = 3 # 沿着y轴堆叠的次数 array_1 = array.copy()[0:repeat_time] array_final, array_final1 = np.meshgrid(array, array_1) print(array_final) """ result: [[1 2 3 4 5] [1 2 3 4 5] [1 2 3 4 5]] """
当然了,还有其他方法,比如 np.vstack() 以及 np.concatenate() 函数均可以实现这个操作。关于这两个函数,可以查看博客------np.concatenate()函数和np.vstack()函数。
以上就是Python如何生成多行重复数据的详细内容,更多请关注php中文网其它相关文章!
python怎么学习?python怎么入门?python在哪学?python怎么学才快?不用担心,这里为大家提供了python速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号