尝试在 while 循环中调用二维数组中的元素（python/numpy）

Question

我正在尝试创建一个程序，将任何给定的矩阵减少为减少的行梯形形式。

我想要的是将行中的每个条目除以前导数字。例如，假设我有：

[ [3, 4, 5], [ 1, 2, 3] ]  # a 2-d array. which gives:

[ 3, 4, 5]
[ 1, 2, 3]

我正在尝试将其转换为 [ 1 , 4/3, 5/3] 。但是，如果您遵循我的代码：

def npgauss(a):
    n = 0

    for i in a[0]:
        x = a[0][0] 
        a[0][n] = a[0][n]/x 
        print a[0][n]
        n = n + 1

您会发现我的问题是第一个条目被自身除以 1，然后当它转到第二个条目时，它将除以新的数字 1 而不是原始数字 1 3号（这就是我想要的）。

我想不出解决办法！我该如何使它除以原始数字？

Answer 1

由于问题提到了 numpy，这里有一个不使用循环的解决方案：

import numpy
a = numpy.array([[3.,4.,5.],[1.,2.,3.]])
a = a/a[:,0:1]
print a

给出所需的结果

array([[ 1.        ,  1.33333333,  1.66666667],
       [ 1.        ,  2.        ,  3.        ]])

这里有一些微妙之处：

1. 我使用浮点数而不是整数，以避免在使用 Python 2 时整数除法出现意外
2. 你必须使用 a [:,0:1] 而不是 a[:,0] 来选择第一列
3. 你不能写 a /= a[:,0:1] 因为就地运算符认为由于某种原因数组广播不同

Answer 2

如果您希望在整个循环中固定 x ，只需在进入循环之前设置其值即可：

x = a[0][0] 
for i in a[0]:
    a[0][n] = a[0][n]/x 
    print a[0][n]
    n = n + 1

Answer 3

您可以使用 列表推导式 来完成此操作

a = [[3, 4, 5], [1, 2, 3]]

b = [[float(i)/row[0] for i in row] for row in a]