通过理解创建NumPy矩阵



假设我有两个向量,xm条目,yn条目。我想要一个numpym x n矩阵a,其中

A[i][j]== x[i]**y[j]

对于这样的问题,我知道如何创建一个具有列表理解能力的2D数组,但有没有一种方法可以使用numpy有效地做到这一点?感谢

如果你正在寻找外部力量,你可以做:

A=np.power.outer(x,y)

(假设导入numpy为np(

您可以使用numpy.outer来执行此

import numpy as np
x = np.arange(4) # size m
y = np.arange(5) # size n
outer_product = np.outer(x,y) # size m x n
print(outer_product)

输出:

[[ 0  0  0  0  0]
[ 0  1  2  3  4]
[ 0  2  4  6  8]
[ 0  3  6  9 12]]

最新更新