Lọc danh sách theo chỉ mục mục?

Question

val data = List("foo", "bar", "bash") 
val selection = List(0, 2) 
val selectedData = data.filter(datum => selection.contains(datum.MYINDEX)) 
//             INVALID CODE HERE^
// selectedData: List("foo", "bash") 

Giả sử tôi muốn lọc List được cung cấp danh sách các chỉ mục đã chọn. Nếu, trong phương thức filter, tôi có thể tham chiếu chỉ mục của một mục danh sách sau đó tôi có thể giải quyết vấn đề này như trên, nhưng datum.MYINDEX không hợp lệ trong trường hợp trên.

Làm cách nào để thay thế?

Answer 1

Cách sử dụng zipWithIndex để giữ tham chiếu đến chỉ mục của mục, lọc như vậy, sau đó ánh xạ chỉ mục đó đi?

data.zipWithIndex 
    .filter{ case (datum, index) => selection.contains(index) } 
    .map(_._1) 

Answer 2

giải pháp đầu tiên mà đến tâm trí của tôi là tạo ra một danh sách các cặp (element, index), lọc mọi phần tử bằng cách kiểm tra nếu lựa chọn chứa chỉ số, sau đó bản đồ danh sách kết quả để giữ chỉ elementd thô (chỉ mục bỏ qua). Mã là tự giải thích:

data.zipWithIndex.filter(pair => selection.contains(pair._2)).map(_._1) 

hoặc dễ đọc hơn:

val elemsWithIndices = data.zipWithIndex 
val filteredPairs = elemsWithIndices.filter(pair => selection.contains(pair._2)) 
val selectedElements = filteredPairs.map(_._1) 

Answer 3

Đó là gọn gàng để làm điều đó theo cách khác về (mặc dù có khả năng chậm với danh sách như đánh chỉ số là chậm (O (n)) Vectors. sẽ tốt hơn. Mặt khác, các contains của giải pháp khác cho mỗi mục trong data là không chính xác nhanh)

val data = List("foo", "bar", "bash") 
     //> data : List[String] = List(foo, bar, bash) 
val selection = List(0, 2) 
     //> selection : List[Int] = List(0, 2) 
selection.map(index=>data(index)) 
     //> res0: List[String] = List(foo, bash) 

Answer 4

vì bạn có một danh sách các chỉ số alread y, cách hiệu quả nhất là phải chọn những chỉ số trực tiếp:

val data = List("foo", "bar", "bash") 
val selection = List(0, 2) 
val selectedData = selection.map(index => data(index)) 

hoặc thậm chí:

val selectedData = selection.map(data) 

hoặc nếu bạn cần để giữ gìn trật tự của các mục trong dữ liệu:

val selectedData = selection.sorted.map(data) 

CẬP NHẬT

Trong tinh thần tìm tất cả các tiện ích thuật toán điện tử, đây là phiên bản sử dụng collect:

val selectedData = data 
    .zipWithIndex 
    .collect { 
    case (item, index) if selection.contains(index) => item 
    } 

Answer 5

này Tác phẩm:

val data = List("foo", "bar", "bash") 
val selection = List(0, 2) 
val selectedData = data.filter(datum => selection.contains(data.indexOf(datum))) 
println (selectedData) 

đầu ra: Danh sách (foo, bash)

Answer 6

Sau đây là cách có lẽ là khả năng mở rộng nhất để làm điều đó về hiệu quả, và không giống như nhiều câu trả lời trên SO, thực sự theo hướng dẫn chính thức về phong cách scala chính xác.

import scala.collection.immutable.HashSet 

val selectionSet = new HashSet() ++ selection 

data.zipWithIndex.collect { 
    case (datum, index) if selectionSet.contains(index) => datum 
} 

Nếu bộ sưu tập kết quả là để được thông qua để thêm map, flatMap, vv, đề nghị chuyển data thành một chuỗi lười biếng. Trong thực tế có lẽ bạn nên làm điều này anyway để tránh 2-pass, một cho zipWithIndex một cho collect, nhưng tôi nghi ngờ khi một điểm chuẩn sẽ đạt được nhiều.