gpt4 book ai didi

python - 你能通用地编写 numpy 切片吗?

转载 作者:行者123 更新时间:2023-12-01 03:38:05 25 4
gpt4 key购买 nike

我想做类似的事情

x[i, :, :] = (rhs[i, :, :]-diag[i] * x[i+1, :, :])/diag[i]

其中 x 和 rhs 是大小为 (T,L,S) 的 3D numpy 数组。 diag 是大小为 T 的一维数组。

这将正确广播。

但现在我想编写一个类似的函数来处理二维数组或其他维数。我怎样才能通用地编写这个代码,以便它可以在任何具有大小为 T 的第一维的数组上工作。我不想仅使用不同数量的冒号来重复代码,因为函数中有很多此类行。

最佳答案

x[i] = (rhs[i] - diag[i] * x[i+1])/diag[i]

那些冒号是完全没有必要的。

关于python - 你能通用地编写 numpy 切片吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40091727/

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