gpt4 book ai didi

python - Numpy 逐元素乘法,多列

转载 作者:行者123 更新时间:2023-11-28 22:09:14 26 4
gpt4 key购买 nike

<分区>

我有以下 numpy 代码。我有一个包含 3d 点的数组 (a) 和另一个包含权重的数组 (b)。我需要将 a 中的每一行乘以相应行中 b 中的每个权重。我希望使这段代码更易于理解并消除循环。

a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.array([[7, 8, 9, 10], [11, 12, 13, 14]])
c = np.zeros((2, 4, 3))

for i, row in enumerate(b):
for j, col in enumerate(row):
print('Mult:', a[i, :], '*', col)
c[i, j, :] = a[i, :] * col

print(c[0, :, :])
print(c[1, :, :])

这是输出。

Mult: [1 2 3] * 7
Mult: [1 2 3] * 8
Mult: [1 2 3] * 9
Mult: [1 2 3] * 10
Mult: [4 5 6] * 11
Mult: [4 5 6] * 12
Mult: [4 5 6] * 13
Mult: [4 5 6] * 14
[[ 7. 14. 21.]
[ 8. 16. 24.]
[ 9. 18. 27.]
[10. 20. 30.]]
[[44. 55. 66.]
[48. 60. 72.]
[52. 65. 78.]
[56. 70. 84.]]

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