gpt4 book ai didi

python - 任意尺寸的 Numpy 切片

转载 作者:IT老高 更新时间:2023-10-28 21:37:03 26 4
gpt4 key购买 nike

我想对一个 numpy 数组进行切片以获得最后一维中的第 i-th 索引。对于 3D 数组,这将是:

slice = myarray[:, :, i]

但我正在编写一个函数,我可以在其中获取任意维度的数组,因此对于 4D 数组,我需要 myarray[:, :, :, i] 等等。有没有一种方法可以在无需显式写入数组维度的情况下为任何数组获取此切片?

最佳答案

...Ellipsis,正是这样做的:

slice = myarray[..., i]

Ellipsis 是 Python 对象,如果您想在方括号表示法之外使用它。

关于python - 任意尺寸的 Numpy 切片,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/12116830/

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