F#过滤子列表

时间:2018-03-04 01:04:32

标签: list filter f#

以下是输入列表:[[10;2;10]; [10;50;10]; [10;1;10]]

如何过滤每个子列表的第二个元素?

下面是我的代码,当我输出结果时,我得到[[10;50;10]],但我想要的是[2;50;1]。反正有没有修复我的代码?我真的想了解F#。感谢您的帮助。

let sub =

     let input = [[10;2;10]; [10;50;10]; [10;1;10]]
     let findIndex input elem = input |> List.findIndex ((=) elem)
     let q = input |> List.filter(fun elem -> findIndex input elem = 1)
     printfn "%A" q

2 个答案:

答案 0 :(得分:1)

以下内容将获得预期结果:

let second (x:List<int>) = x.[1]
let q = List.map second input 

List.map是一个更高阶的函数,它通过将一个函数(第一个参数,这里是返回列表的第二个元素的函数second)应用于列表来生成一个新列表(这里{ {1}}):

input

答案 1 :(得分:1)

使用List.map,而不是List.filter

List.filter根据您提供的功能保留/删除输入列表中的项目,例如: myList只保留val filter: ('a -> bool) -> 'a list -> 'a list中的偶数。您可以根据其类型签名({1}}了解此功能,这意味着它需要一个函数(接受一个'a并返回一个布尔值:('a -> bool)),然后获取一个列表,并返回相同类型的列表。

另一方面,

List.map根据您提供的函数将列表的每个元素转换为您想要的任何元素。在你的情况下你会像这样使用它:

let input = [[10;2;10]; [10;50;10]; [10;1;10]]
let result = input |> List.map (fun numbers -> numbers.[1])
printfn "%A" result

List.map的签名是val map: ('a -> 'b) -> 'a list -> 'b list,这意味着它需要一个将'a映射到'b的函数(在您的情况下,这将映射int list转到int s),获取第一件事的列表,并返回第二件事的列表。

使用List.map List.tryIndex

请注意,如果任何子列表太短,程序将崩溃。如果这是一个问题,您可以使用myList.[i]的安全版本;即List.tryIndex,如果找不到该项,则返回None。试试这个:

// Note the last sublist
let input = [[10;2;10]; [10;50;10]; [10;1;10]; [-1]]
let result : int option list = input |> List.map (fun numbers -> List.tryIndex 1 numbers)
printfn "%A" result
// This prints:
// [Some 2; Some 50; Some 1; None]