Có thể trong NumPy để sử dụng danh sách cắt tiên tiến và vẫn nhận được một cái nhìn?

Question

Nói cách khác, tôi muốn làm một cái gì đó giống như

A[[-1, 0, 1], [2, 3, 4]] += np.ones((3, 3)) 

thay vì

A[-1:3, 2:5] += np.ones((1, 3)) 
A[0:2, 2:5] += np.ones((2, 3)) 

Answer 1

Nếu tôi hiểu đúng, bạn có thể làm những gì bạn muốn làm gì với những điều sau đây:

A[[[-1],[0],[1]],[2,3,4]] += np.ones((3, 3)) 

Tuy nhiên, các folks numPy làm một chức năng, ix_, để làm cho nó một chút dễ dàng hơn:

A[np.ix_([-1,0,1],[2,3,4])] += np.ones((3, 3)) 

Tôi hy vọng rằng sẽ giúp.