首頁猿問用(n-1)d數(shù)組索引n維數(shù)組

用(n-1)d數(shù)組索引n維數(shù)組

Python

慕妹3242003 2019-07-12 15:27:00

用(n-1)d數(shù)組索引n維數(shù)組使用(n-1)維數(shù)組訪問給定維度的n維數(shù)組的最優(yōu)雅方法是什么，如虛擬示例中所示。a = np.random.random_sample((3,4,4))b = np.random.random_sample((3,4,4))idx = np.argmax(a, axis=0)我現(xiàn)在如何訪問idx a使極大值進入a好像我用過a.max(axis=0)？或如何檢索idx在……里面b?我想用np.meshgrid但我覺得這太過分了。注意尺寸axis可以是任何有用的軸(0，1，2)并且事先不知道。有什么好辦法嗎？

查看完整描述

3 回答

臨摹微笑

TA貢獻1982條經驗獲得超2個贊

all_idx很好。我沒有意識到使用argmax輸出的形狀而不是原始數(shù)組的形狀會簡化事情。至于元組而不是列表，高級索引語義與元組相比更加清晰。在這種情況下，列表的行為碰巧是一樣的，這是因為在特定條件下，向后兼容性處理將列表轉換為元組(但沒有得到完全正確的記錄)。在NumPy索引中，當一個列表被當作一個元組對待，當一個列表被當作一個數(shù)組處理時，這可能會令人驚訝，所以我更喜歡顯式地創(chuàng)建這個元組。

反對回復 2019-07-12