gpt4 book ai didi

python-2.7 - 如何使用广播从具有列表 2d 索引的 2D numpy 数组中获取元素?

转载 作者:行者123 更新时间:2023-12-03 10:03:00 26 4
gpt4 key购买 nike

<分区>

如果我有一个 2D numpy 数组,我想使用行列索引对列表提取元素。

xy = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
idx = np.array([[0, 0], [1, 1], [2, 2]])

for循环解决方案:

elements = list()
for i in idx:
elements.append(xy[idx[i][0], xy[idx[i][1])

输出:

print(elements)
>> [1, 5, 9]

如果 idx 是一个元组列表,我找到了解决方案,但我希望有一个无需先将 idx 转换为元组的解决方案。

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