gpt4 book ai didi

python - 在 numpy 中计算对角线和

转载 作者:行者123 更新时间:2023-12-02 01:51:42 32 4
gpt4 key购买 nike

<分区>

我有一个相当大的矩形 numpy 数组,形状为 (m, n),例如:

>>> a.shape
(27584, 34092)

我必须计算每个 anti-diagonal 的总和的阵列。这个新数组的形状为 (m + n - 1,)

简单的方法是:

m, n = a.shape
r = np.zeros(m + n - 1)
for i in range(m):
for j in range(n):
r[i + j] += a[i][j]

# r is the sum of all anti-diagonals of a

这显然很慢,有什么方法可以使用巧妙的 numpy 原语来执行计算吗?我唯一的其他选择是用 C++ 编写代码,这也是可行的 - 但需要更多工作。

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