Python 生成多行重复数据的方法实现

目录

• 引言
• 一般方法
• 使用np.repeat函数
• 使用np.meshgrid函数

引言

在做科学计算或者模拟仿真的时候，相信不少小伙伴会遇到这样的问题，比如，我们有一个一维数组如下所示：

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函数

显然，上述方法比较麻烦，为了简化，我们可以世界使用 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()函数。

使用np.meshgrid函数

当然了，对于这种情况，最简单的方法就是使用 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 生成多行重复数据的方法实现的文章就介绍到这了,更多相关Python 生成多行重复数据的方法实现内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！