开发者社区 问答 正文

如何在python中简化二维numpy数组?

给定一个3乘3 numpy数组

a = numpy.arange(0,27,3).reshape(3,3)

array([[ 0, 3, 6],

[ 9, 12, 15],

[18, 21, 24]])

为了规范二维数组的行,我想到了

row_sums = a.sum(axis=1) # array([ 9, 36, 63]) new_matrix = numpy.zeros((3,3)) for i, (row, row_sum) in enumerate(zip(a, row_sums)): new_matrix[i,:] = row / row_sum 必须有更好的方法,不是吗?

可能需要澄清:通过标准化我的意思是,每行条目的总和必须为1。但是我认为这对于大多数人来说都是显而易见的。 问题来源于stack overflow

展开
收起
保持可爱mmm 2020-02-08 14:21:59 553 分享 版权
1 条回答
写回答
取消 提交回答
  • row_sums = a.sum(axis=1) new_matrix = a / row_sums[:, numpy.newaxis] row_sums[:, numpy.newaxis]重塑row_sums从存在(3,)到存在(3, 1)。当你这样做a / b,a并b会相互播出。

    2020-02-08 14:22:17
    赞同 展开评论