gpt4 book ai didi

python - 添加两个大小不等的 numpy 数组(长度 n 和 m)以生成 n * m 数组而不使用 for 循环

转载 作者:太空宇宙 更新时间:2023-11-03 14:44:41 45 4
gpt4 key购买 nike

<分区>

我正在尝试向量化几个简单的计算。我有两个数组:

A = np.array([1,2,3,4])
B = np.array([1,2])

我想计算一个数组C,计算如下:

C = []
for i in A:
for j in B:
C.append(i+j)

C = np.array(C)

这样

C = array([2, 3, 3, 4, 4, 5, 5, 6])

实际上数组非常大,这些 for 循环需要相当长的时间。有没有办法将这些计算向量化?

45 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com