假設我有一個數組 a，它可以是 N 維的，但為了簡單起見，讓我們說它：a = np.array([[0,1,2],
[3,4,5]])我也有一個長度為 N 的面具或作物（我不知道正確的術語），所以在這種情況下是這樣的：b = [1,2]我如何“裁剪”數組，使其：a = a[:b]這將切片與：a = a[:1,:2]但不管 N 的值如何都會起作用。
1 回答
慕尼黑的夜晚無繁華
TA貢獻1864條經驗 獲得超6個贊
使用slice符號 -
a[tuple([slice(None,bi) for bi in b])]
或者用它的簡寫np.s_——
a[tuple([np.s_[:bi] for bi in b])]
添加回答
舉報
0/150
提交
取消