在python numpy中为任何n构建一个nxn矩阵

时间:2014-04-03 21:59:58

标签: python arrays numpy matrix

是否可以使用numpy for python(版本3.3)编写用于构建nxn矩阵的代码,而不指定n?我需要将条目索引为A_i,j或类似的东西,但我甚至不知道如何定义A_i,j以便它们实际上是对象。我觉得这样的事情可能有用:

n    
i=1
j=1

when i (is less than) n+1

  when j (is less than) i+1
   A_i,j= f(i,j)
   j+=1

i+=1

但这不起作用......有什么建议吗? 我的最终目标是为任意nxn矩阵编写QR分解。但我需要知道如何定义我首先要处理的矩阵。我是python的新手,因此numpy因此不知道任何东西。任何帮助将不胜感激。我也是stackexchange的新手,很抱歉我有那么糟糕的代码。 (小于)应该是指三角形标志缺少基部,头部指向左侧 - 这明显小于符号

1 个答案:

答案 0 :(得分:3)

您可以创建一个空的n x n数组:

import itertools
import numpy as np

my_array = np.empty([n, n])

然后将坐标i, j的值设置为f(i, j)

for i, j in itertools.product(range(n), range(n)):
    my_array[i, j] = f(i, j)