Numpy: arr [..., 0 ,:] hoạt động. Nhưng làm thế nào để lưu trữ dữ liệu chứa trong lệnh slice (..., 0, :)?

Question

Trong Numpy (và Python nói chung, tôi giả sử), làm thế nào để lưu trữ một slice-index, chẳng hạn như (..., 0, :), để vượt qua nó và áp dụng nó cho các mảng khác nhau? Nó sẽ được tốt đẹp để, nói, có thể vượt qua một slice-index đến và đi từ chức năng.

Answer 1

Python tạo các đối tượng đặc biệt ngoài cú pháp slice, nhưng chỉ bên trong dấu ngoặc vuông để lập chỉ mục. Bạn có thể tạo ra các đối tượng bằng tay (trong trường hợp này, (...,0,:) là (Ellipsis, 0, slice(None, None, None)), hoặc bạn có thể tạo một đối tượng helper nhỏ:

class ExtendedSliceMaker(object): 
    def __getitem__(self, idx): 
     return idx 

>>> ExtendedSliceMaker()[...,0,:] 
(Ellipsis, 0, slice(None, None, None)) 

Answer 2

Tương đương với (...,0,:) nên ...

>>> myslice = (..., 0, slice(None, None)) 
>>> myslice 
(Ellipsis, 0, slice(None, None, None)) 

Answer 3

Tôi nghĩ rằng bạn muốn chỉ làm myslice = lát (1,2) để ví dụ xác định một lát sẽ trở lại các yếu tố thứ 2 (tức là myarray [myslice] == myarray [1: 2])

Answer 4

Điều gọn về Python là bạn thực sự có thể làm cho một lớp học để kiểm tra cách thức các Python sử dụng phương pháp ma thuật __getitem__ để xử lý các hoạt động lập chỉ mục, vì vậy chúng tôi sẽ tạo một lớp quá tải để hiển thị cho chúng tôi những gì đã được truyền vào, khởi tạo lớp và "lập chỉ mục" cho cá thể:

class foo: 
    def __getitem__(self, index): print index 

foo()[...,0,:] 

Và kết quả của chúng tôi là:

(Ellipsis, 0, slice(None, None, None)) 

Ellipsis và slice đang dựng sẵn, và chúng ta có thể đọc tài liệu của họ:

help(Ellipsis) 
help(slice) 

Answer 5

Sử dụng s_ trong NumPy:

In [1]: np.s_[...,0,:] 
Out[1]: (Ellipsis, 0, slice(None, None, None))