快捷導(dǎo)航

np.repeat()函數(shù)的具體使用

更新時間：2023年03月13日 09:44:22 作者：勤奮的大熊貓

本文主要介紹了np.repeat()函數(shù)的具體使用，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧

使用np.repeat()展平二維數(shù)組

代碼如下：

import numpy as np


class Debug:
? ? def __init__(self):
? ? ? ? self.array1 = np.array([[1, 2], [3, 4]])

? ? def mainProgram(self):
? ? ? ? print("The value of array1 is: ")
? ? ? ? print(self.array1)
? ? ? ? print("The repeated array is: ")
? ? ? ? array2 = np.repeat(self.array1, repeats=1)
? ? ? ? print(array2)


if __name__ == '__main__':
? ? main = Debug()
? ? main.mainProgram()
"""
The value of array1 is:?
[[1 2]
?[3 4]]
The repeated array is:?
[1 2 3 4]
""" ? ?

我們可以看到我們輸入的是一個二維數(shù)組，當(dāng)我們設(shè)定 repeats 值為 1 時，輸出結(jié)果變成了一個一維數(shù)組，因此這時的 np.repeats 函數(shù)類似numpy.ndarray.flatten()函數(shù)的功能。

np.repeat()函數(shù)的坐標(biāo)軸問題

接下來我們研究一下關(guān)于 axis 坐標(biāo)軸的問題。

二維

對于數(shù)組是二維的情況，代碼如下：

import numpy as np


class Debug:
? ? def __init__(self):
? ? ? ? self.array1 = np.array([[1, 2], [3, 4]])

? ? def mainProgram(self):
? ? ? ? print("The value of array1 is: ")
? ? ? ? print(self.array1)
? ? ? ? print("The array2 is: ")
? ? ? ? array2 = np.repeat(self.array1, repeats=2, axis=0)
? ? ? ? print(array2)
? ? ? ? print("The array3 is: ")
? ? ? ? array3 = np.repeat(self.array1, repeats=2, axis=1)
? ? ? ? print(array3)


if __name__ == '__main__':
? ? main = Debug()
? ? main.mainProgram()
"""
The value of array1 is:?
[[1 2]
?[3 4]]
The array2 is:?
[[1 2]
?[1 2]
?[3 4]
?[3 4]]
The array3 is:?
[[1 1 2 2]
?[3 3 4 4]]
"""

我們可以看到，axis=0 時表示沿著y方向重復(fù)，axis=1 時表示沿著x方向重復(fù)。我們可以對比numpy數(shù)組的坐標(biāo)軸表示，二維時，坐標(biāo)軸為 (y, x)，從左向右第一個參數(shù) 0 便代表 y 軸，1 代表 x軸。

三維

接下來我們研究一下數(shù)組是三維的情況，代碼如下：

import numpy as np


class Debug:
? ? def __init__(self):
? ? ? ? self.array1 = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

? ? def mainProgram(self):
? ? ? ? print("The value of array1 is: ")
? ? ? ? print(self.array1)
? ? ? ? print("The array2 is: ")
? ? ? ? array2 = np.repeat(self.array1, repeats=2, axis=0)
? ? ? ? print(array2)
? ? ? ? print("The array3 is: ")
? ? ? ? array3 = np.repeat(self.array1, repeats=2, axis=1)
? ? ? ? print(array3)
? ? ? ? print("The array4 is: ")
? ? ? ? array4 = np.repeat(self.array1, repeats=2, axis=2)
? ? ? ? print(array4)


if __name__ == '__main__':
? ? main = Debug()
? ? main.mainProgram()
"""
The value of array1 is:?
[[[1 2]
? [3 4]]

?[[5 6]
? [7 8]]]
The array2 is:?
[[[1 2]
? [3 4]]

?[[1 2]
? [3 4]]

?[[5 6]
? [7 8]]

?[[5 6]
? [7 8]]]
The array3 is:?
[[[1 2]
? [1 2]
? [3 4]
? [3 4]]

?[[5 6]
? [5 6]
? [7 8]
? [7 8]]]
The array4 is:?
[[[1 1 2 2]
? [3 3 4 4]]

?[[5 5 6 6]
? [7 7 8 8]]]
"""

我們可以看到，axis=0 對應(yīng)與沿著z軸重復(fù)，axis=1 對應(yīng)沿著y軸重復(fù)，axis=2 對應(yīng)沿著x軸重復(fù)。對比numpy坐標(biāo)軸的表示，我們知道三維坐標(biāo)軸為 (z, y, x)，所以從左向右，0 對應(yīng)z軸，1 對應(yīng) y軸，2 對應(yīng) x 軸。

到此這篇關(guān)于np.repeat()函數(shù)的具體使用的文章就介紹到這了,更多相關(guān)np.repeat()函數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: