python: trích xuất một lát của một mảng đa chiều cho chỉ số thứ nguyên

Question

Tôi biết cách lấy x[:,:,:,:,j,:] (lấy phần thứ j của thứ nguyên 4).

Có cách nào để làm điều tương tự nếu thứ nguyên được biết trong thời gian chạy và không phải là hằng số đã biết không?

Answer 1

Bạn có thể sử dụng slice chức năng và gọi nó với các danh sách biến thích hợp trong thời gian chạy như sau:

# Store the variables that represent the slice in a list/tuple 
# Make a slice with the unzipped tuple using the slice() command 
# Use the slice on your array 

.210

Ví dụ:

>>> from numpy import * 
>>> a = (1, 2, 3) 
>>> b = arange(27).reshape(3, 3, 3) 
>>> s = slice(*a) 
>>> b[s] 
array([[[ 9, 10, 11], 
     [12, 13, 14], 
     [15, 16, 17]]]) 

Đây là giống như:

>>> b[1:2:3] 
array([[[ 9, 10, 11], 
     [12, 13, 14], 
     [15, 16, 17]]]) 

Cuối cùng, tương đương với không chỉ định bất cứ điều gì giữa 2 : trong cách viết thông thường là đặt None ở những nơi trong tuple bạn tạo nên.

Answer 2

Một lựa chọn để làm như vậy là để xây dựng cắt programatically:

slicing = (slice(None),) * 4 + (j,) + (slice(None),) 

Một cách khác là sử dụng numpy.take() hay ndarray.take():

>>> a = numpy.array([[1, 2], [3, 4]]) 
>>> a.take((1,), axis=0) 
array([[3, 4]]) 
>>> a.take((1,), axis=1) 
array([[2], 
     [4]]) 

Answer 3

Bạn cũng có thể sử dụng dấu ba chấm để thay thế dấu hai chấm lặp lại. Xem answer đến How do you use the ellipsis slicing syntax in Python? để biết ví dụ.