numpy 数组问题请大神解答

课后作业？建议先自己研究下

比较笨的办法：

a = np.random.randint(0, 10, (5, 5)) # 随机生成一个 5x5 的整数矩阵

b = [(j, i//a0.shape[1]) for i,j in enumerate(a0.flatten())] # 拍扁之后生成列表，每个元素为 (数值，行数）

c = [i[1] for i in sorted(b0, key=lambda x: x[0]]

就是感觉数据量大的话，这么写会很慢……

对 a 做个遍历将值和下标做成一个 dict，然后拍平后再对 b 做一个遍历处理，复杂度 O(n)

@jiejiss 看着确实像，其实不是，我想问有没有快速现成的方法，比如通过矩阵相乘。

@volvo007 我现在要处理的元素数量是 150 * 50000 个

@renmu123 有没有通过调用 numpy 或者 pytorch 函数的方法

@jiejiss @renmu123 @volvo007 谢谢大家已经解决了，昨天脑子比较混沌

@Porphet #4 拿 C 写出来，编译成动态链接库然后让 Python 用 ctypes 之类的去调用吧……

涉及到排序还有查下标，那 Python 再怎么优化也一定是比较慢的

@jiejiss 这是极致的优化了，调用 numpy.argsort()得到排序后的下标，然后除一下 dim=1 的长度就可以了

@Porphet 哦，明白你意思了。昨天也想到了这个函数，但是一时没想起来后面怎么处理