开发者

Does F# have an equivalent to Haskell's take?

In Haskell, there is a function "take n list" which returns the first n elements from a list. For example "sum (take 3 xs)" sums up the first three elements in the list xs. Does F开发者_高级运维# have an equivalent? I expected it to be one of the List-functions, but I can't spot anything that seems to match.


To clarify a few things, the difference between Seq.take and Seq.truncate (as pointed out by @sepp2k) is that the second one will give you a sequence that returns at most the number of elements you specified (but if the length of the sequence is less, it will give you less elements).

The sequence generated Seq.take function will throw an exception if you try to access an element beyond the length of the original list (Note that the Seq.take function doesn't throw the exception immediately, because the result is lazily generated sequence).

Also, you don't need to convert the list to a sequence explicitly. Under the cover, list<'a> is a .NET class that inherits from the seq<'a> type, which is an interface. The type seq<'a>is actually just a type alias for IEnumerable<'a>, so it is implemented by all other collections (including arrays, mutable lists, etc.). The following code will work fine:

let list = [ 1 .. 10 ]
let res = list |> Seq.take 5

However, if you want to get a result of type list<int> you'll need to convert sequence back to a list (because a list is more specific type than a sequence):

let resList = res |> List.ofSeq

I'm not sure why F# libraries don't provide List.take or List.truncate. I guess the goal was to avoid reimplementing the whole set of functions for all types of collections, so those where the implementation for sequences is good enough when working with a more specific collection type are available only in the Seq module (but that's only my guess...)


Yeah, it's called Seq.take. Usage seems to be identical to Haskell's: Seq.take count source. To use it on a list, use List.toSeq first. (Update: Apparently from the comments, this isn't necessary.)


Seq.take works, as others already have stated, but all Seq operations on a list come at a cost. In the case of Seq.take, it's not surprising, as the list has to be copied.

It's more noteworthy that, for example, Seq.concat on a list takes a lot more time than List.concat. I suppose that implies that you don't just access the list as a seq when you call a Seq.xxx function, but that the list in copied/converted to a Seq behind the scenes as well.

edit: The reason I drew the conclusion above, was this bench using F# interactive:

#time "on";;
let lists = [for i in 0..5000000 -> [i..i+1]];;
Seq.length (Seq.concat lists);;
List.length (List.concat lists);;

On my machine, the List.length version takes about 1.9 secs, whereas the Seq.length version takes about 3.8 secs (shortest time of a few repeated tests of the length lines only, excluding the list generation line).


Elaborating on @Jonathan Tyson comment

List.take is now part of the standard F# library and it is recommended to use that instead of the Seq.Take

The following should work

List.take <count>

Also, see this answer to understand the difference between take and truncate

0

上一篇:

下一篇:

精彩评论

暂无评论...
验证码 换一张
取 消

最新问答

问答排行榜