在Python/numpy的

Question

排序容器我需要排序特殊numpy的陣列，其中，尺寸19的塊構成的元件，使用用戶定義的函數來確定這樣的塊的值。

第一次嘗試一直包裹陣列中的類和重載[]操作：

class W: 
    def __init__(self, filename="nn.txt"): 
     self.nn = array([int(i) for i in open(filename, "r").readlines()[1:]]) 
     self.size = self.nn.size/19 

    def __getitem__(self, idx): 
     return self.nn[idx:idx+19] 

    def __len__(self): 
      return self.size 

使用這種結構I提供一個比較操作符，其被傳遞給排序（）：

def avg_cmp(x, y): 
    return int(average(x)) - int(average(y)) 

u = W("nnsmall.txt") 
sorted(u, cmp=avg_cmp) 

然而，這種做法是太慢了。

任何提示？

Answer 1

你試過sorted(u, key=average)？這隻會計算一次每列的平均值。

如果數組的大小始終是整除19餘無：

>>> import numpy as np 
>>> n = 2 
>>> u = np.array([v for v in range(19*n)]) 

>>> u = u.reshape(n,19) 

>>> sorted(c, key=np.average) 

[array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]), 
array([19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37])]