首页 > 后端开发 > Python教程 > Python如何生成多行重复数据

Python如何生成多行重复数据

PHPz
发布: 2023-05-11 13:16:13
转载
1768 人浏览过

引言

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

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如何生成多行重复数据的详细内容。更多信息请关注PHP中文网其他相关文章!

相关标签:
来源:yisu.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板