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()函数。

原文地址：https://blog.csdn.net/u011699626/article/details/129461920