Ký hiệu slice NumPy trong từ điển

Question

Tôi tự hỏi liệu có thể lưu trữ ký hiệu slice numpy trong từ điển python hay không. Một cái gì đó như:

lookup = {0:[:540], 
      30:[540:1080], 
      60:[1080:]} 

Có thể sử dụng cú pháp lát trăn gốc, ví dụ: slice(0,10,2), nhưng tôi đã không thể lưu trữ các lát phức tạp hơn. Ví dụ, một cái gì đó là đa chiều [:,:2,:, :540].

Công việc hiện tại của tôi xung quanh là lưu trữ các giá trị dưới dạng bộ dữ liệu và sau đó giải nén chúng thành các lát cần thiết.

Làm việc bằng Python 2.x.

Answer 1

Cú pháp [:, :2, :, :540] được biến thành một tuple của slice đối tượng bằng Python:

(slice(None, None, None), 
slice(None, 2, None), 
slice(None, None, None), 
slice(None, 540, None)) 

Một cách thuận tiện để tạo tuple này là sử dụng các chức năng đặc biệt * np.s_. Bạn chỉ cần vượt qua biểu thức [...]. Ví dụ:

>>> np.s_[:540] 
slice(None, 540, None) 
>>> np.s_[:, :2, :, :540] 
(slice(None, None, None), 
slice(None, 2, None), 
slice(None, None, None), 
slice(None, 540, None)) 

Rồi từ điển của bạn lát có thể được viết như sau:

lookup = {0: np.s_[:540], 
      30: np.s_[540:1080], 
      60: np.s_[1080:]} 

* kỹ thuật s_ là một bí danh cho lớp IndexExpression mà thực hiện một phương pháp đặc biệt __getitem__.

Answer 2

NumPy có rất nhiều Indexing routines .Và trong trường hợp này bạn có thể sử dụng các chức năng sau cho Tạo mảng chỉ mục:

c_: Dịch vật lát để nối dọc theo trục thứ hai.

r_: Dịch các đối tượng lát để nối dọc theo trục đầu tiên.

s_: Cách tốt hơn để tạo bộ chỉ mục cho mảng.

Bạn cũng có thể sử dụng numpy.unravel_index:

Chuyển đổi một tuple của mảng chỉ số vào một loạt các chỉ số căn hộ, áp dụng chế độ ranh giới để đa-index.