在索引它們時是否可以對 numpy.array 應用幾個條件？就我而言，我想在第 5 步中顯示前 10 個元素，然后顯示 2 個相鄰元素：
numpy.arange(40)
#Output is:
array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33,
34, 35, 36, 37, 38, 39])
將我的條件應用于這個陣列我想得到這個：
array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 14, 15, 20, 21, 26, 27, 32,
33, 38, 39])
我還沒有找到任何解決方案。我認為它應該是這樣的：
np.arange(40)[0:10, 10:len(np.arange(40)):5]
但這對我不起作用。
uj5u.com熱心網友回復：
您可以嘗試對整形陣列進行自定義索引：
n = 40
idx = np.zeros(n//2, dtype=bool)
idx[:5] = True
idx[4:None:3] = True
>>> np.arange(n).reshape(-1,2)[idx]
array([[ 0, 1],
[ 2, 3],
[ 4, 5],
[ 6, 7],
[ 8, 9],
[14, 15],
[20, 21],
[26, 27],
[32, 33],
[38, 39]])
>>> np.arange(n).reshape(-1,2)[idx].ravel()
array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 14, 15, 20, 21, 26, 27, 32,
33, 38, 39])
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/361972.html
上一篇：行內樣式單個React組件