我目前正在使用 python 和 openCV 進行計算機視覺專案。我有一個像這樣的二維 numpy 陣列：
[100 38 18]
[134 332 16]
[136 200 16]
[136 288 15]
[138 160 17]
[138 246 15]
[140 76 12]
[140 116 12]
[142 34 14]
二維陣列已按第一列排序。這作業正常。現在我需要按第二列對 3 行的包進行排序。這是我需要達到的結果：
[100 38 18]
[136 200 16]
[134 332 16]
[138 160 17]
[138 246 15]
[136 288 15]
[142 34 14]
[140 76 12]
[140 116 12]
我怎樣才能做到這一點？
uj5u.com熱心網友回復：
考慮將資料重塑為 3d，然后使用 for 回圈對每個陣列進行排序并轉換回 np.array
np.array([sorted(i, key = lambda x: x[1]) for i in ar.reshape(3, -1, ar.shape[1])]).reshape(ar.shape)
array([[100, 38, 18],
[136, 200, 16],
[134, 332, 16],
[138, 160, 17],
[138, 246, 15],
[136, 288, 15],
[142, 34, 14],
[140, 76, 12],
[140, 116, 12]])
uj5u.com熱心網友回復：
只是通過 NumPy，而不是回圈：
sort_ = np.argsort(np.split(a[:, 1], a.shape[0] // 3))
# [[0 2 1]
# [1 2 0]
# [2 0 1]]
sort_ = np.linspace(0, a.shape[0] - 3, a.shape[0] // 3, dtype=np.int64)[:, None]
# [[0 2 1]
# [4 5 3]
# [8 6 7]]
a = a[sort_.ravel()]
uj5u.com熱心網友回復：
除了使用回圈，我別無他法
讓A成為你的陣列
output = np.zeros((0, 3))
for i in range(int(A.shape[0]/3)):
output = np.vstack((output, A[3*i np.argsort(A[3*i:3*(i 1), 1])]))
注意：我假設您的陣列有許多行是 3 的倍數
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/479492.html