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如何產生多行重複數據的詳細內容。更多資訊請關注PHP中文網其他相關文章！