python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python 生成多行重复数据

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

作者:勤奋的大熊猫

本文主要介绍了Python 生成多行重复数据的方法实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

引言

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

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 生成多行重复数据的方法实现内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文