首頁 > 後端開發 > 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
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板