choose函數實現條件篩選
choose函數主要利用下標以及下標所在的位置對相關的數組進行篩選，看下面的示例:
In [129]: control
Out[129]:
array([[1, 0, 1],
[2, 1, 0],
[1, 2, 2]])
In [130]: choose(control, [10, 11, 12]) # control中，1對應11，0對弈10，2對應12
Out[130]:
array([[11, 10, 11],
[12, 11, 10],
[11, 12, 12]])
--------------------------------------------------------
In [131]: i0
Out[131]:
array([[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])
In [132]: i2
Out[132]:
array([[20, 21, 22],
[23, 24, 25],
[26, 27, 28]])
In [133]: choose(control, [i0, 10, i2]) # control中，第一個元素1對應10，第二個元素[0, 1]0對應i0中的[0, 1]1，以此類推。
Out[133]:
array([[10, 1, 10],
[23, 10, 5],
[10, 27, 28]])
------------------------------------------------------------
In [135]: a = array([[0, 1, 2], [10, 11, 12], [20, 21, 22]])
In [136]: a
Out[136]:
array([[ 0, 1, 2],
[10, 11, 12],
[20, 21, 22]])
In [137]: a < 10
Out[137]:
array([[ True, True, True],
[False, False, False],
[False, False, False]], dtype=bool)
In [138]: choose(a < 10, [a, 10]) # True可以看成是1，對應10；False看成0，對應a中相同位置的元素。
Out[138]:
array([[10, 10, 10],
[10, 11, 12],
[20, 21, 22]])
------------------------------------------------------------
# 下面的例子將數組中所有小于10的值變成了10， 大于15的值變成了15.
In [139]: a
Out[139]:
array([[ 0, 1, 2],
[10, 11, 12],
[20, 21, 22]])
In [140]: a < 10
Out[140]:
array([[ True, True, True],
[False, False, False],
[False, False, False]], dtype=bool)
In [141]: a > 15
Out[141]:
array([[False, False, False],
[False, False, False],
[ True, True, True]], dtype=bool)
In [142]: lt = a < 10
In [143]: a > 15
Out[143]:
array([[False, False, False],
[False, False, False],
[ True, True, True]], dtype=bool)
In [144]: gt = a > 15
In [145]: gt
Out[145]:
array([[False, False, False],
[False, False, False],
[ True, True, True]], dtype=bool)
In [146]: 2 * gt
Out[146]:
array([[0, 0, 0],
[0, 0, 0],
[2, 2, 2]])
In [147]: gt * 2
Out[147]:
array([[0, 0, 0],
[0, 0, 0],
[2, 2, 2]])
In [148]: a
Out[148]:
array([[ 0, 1, 2],
[10, 11, 12],
[20, 21, 22]])
In [149]: lt = a < 10
In [150]: lt
Out[150]:
array([[ True, True, True],
[False, False, False],
[False, False, False]], dtype=bool)
In [151]: gt = a > 15
In [152]: gt
Out[152]:
array([[False, False, False],
[False, False, False],
[ True, True, True]], dtype=bool)
In [153]: choice = lt + 2 * gt
In [154]: choice
Out[154]:
array([[1, 1, 1],
[0, 0, 0],
[2, 2, 2]])
In [155]: choose(choice, [a, 10, 15])
Out[155]:
array([[10, 10, 10],
[10, 11, 12],
[15, 15, 15]])
transpose實現轉置
np.transpose實現數組的轉置，并且可以定制轉置的維度。示例如下：
In [2]: a = np.arange(6).reshape(1, 2, 3)
In [3]: a
Out[3]:
array([[[0, 1, 2],
[3, 4, 5]]])
In [5]: np.transpose(a)
Out[5]:
array([[[0],
[3]],
[[1],
[4]],
[[2],
[5]]])
In [6]: np.transpose(a).shape
Out[6]: (3, 2, 1)
In [7]: np.transpose(a, axes=(1, 2, 0)) # 沿著指定維度的轉置
Out[7]:
array([[[0],
[1],
[2]],
[[3],
[4],
[5]]])
In [8]: np.transpose(a, axes=(1, 2, 0)).shape
Out[8]: (2, 3, 1)
squeeze壓縮數組維度
np.squeeze可以壓縮數組的維度，去掉維度等于1的那一維。示例如下：
In [9]: a
Out[9]:
array([[[0, 1, 2],
[3, 4, 5]]])
In [10]: a.shape
Out[10]: (1, 2, 3)
In [12]: np.squeeze(a)
Out[12]:
array([[0, 1, 2],
[3, 4, 5]])
In [13]: np.squeeze(a).shape
Out[13]: (2, 3)
repeat復制數組元素
np.repeat實現數組元素的復制，并且可以定制維度，示例如下：
In [14]: w = np.array([[-1,-1,-1], [-1,8,-1], [-1,-1,-1]])
In [15]: w.shape
Out[15]: (3, 3)
In [16]: w = w.reshape(1, 1, 3, 3)
In [17]: w
Out[17]:
array([[[[-1, -1, -1],
[-1, 8, -1],
[-1, -1, -1]]]])
In [18]: w.shape
Out[18]: (1, 1, 3, 3)
In [19]: np.repeat(w, 3)
Out[19]:
array([-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 8, 8, 8, -1, -1,
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1])
In [21]: np.repeat(w, 3).shape
Out[21]: (27,)
In [22]: np.repeat(w, 3, axis=1) # 定制復制維度，沿著第一維進行復制
Out[22]:
array([[[[-1, -1, -1],
[-1, 8, -1],
[-1, -1, -1]],
[[-1, -1, -1],
[-1, 8, -1],
[-1, -1, -1]],
[[-1, -1, -1],
[-1, 8, -1],
[-1, -1, -1]]]])
In [23]: np.repeat(w, 3, axis=1).shape
Out[23]: (1, 3, 3, 3)
