从 F# 中的 lambda 表达式返回不同类型的数组

Question

我有一个记录列表，

type Item = { Color : string; Size : int}
let itemList = [{Color="Red"; Size=1};
                {Color="Green"; Size=2};
                {Color="Blue"; Size=3};]

我希望将我的记录列表转换为值数组，例如 [|"Red";"Green";"Blue"|] 或 [|1;2;3|]

我可以像这样到达那里

type ItemType =
| Color of string
| Size of int

type ItemEnum =
| C
| S

let GetProp x y =
match x with
| C -> List.toArray y |> Array.map(fun x -> ItemType.Color(x.Color))
| S -> List.toArray y |> Array.map(fun x -> ItemType.Size(x.Size))

，但是当我调用 GetProp S itemList 时我会回来 [ |尺寸 1；尺寸 2；尺寸 3|]。有用但不完全是我想要的。

我已经尝试过以下方法

let GetProp2 x y : 'a[] =
match x with
| Color -> List.toArray y |> Array.map(fun x -> x.Color)
| Size -> List.toArray y |> Array.map(fun x -> x.Size)

，但它不喜欢两种不同的返回类型。

我愿意接受有关不同（更实用？）方法的建议，并且感谢您的意见。

Answer 1

自定义变体类型确实是此处的方法（通常在您需要“X 或 Y”类型的任何地方）。但是，根据定义，您的函数看起来可能会返回一个混合了 Color 和 Size 的数组，但实际上您似乎只希望它返回一个或另一个。如果是这样，这最好地反映在类型中：

type Items =
| Colors of string[]
| Sizes of int[]

let GetProp x ys =
match x with
| C -> Colors [| for y in ys -> y.Color |]
| S -> Sizes [| for y in ys -> y.Size |]

顺便说一下，这里使用数组作为返回类型，而不是通常的惰性序列（seq）有什么特殊原因吗？

Answer 2

您可以使用活动模式从以下几个角度查看数据：

let (|AsColor|) (v: Item list) = [| for i in v -> i.Color |];;

let (|AsSize|) (v: Item list) = [| for i in v -> i.Size |];;

match itemList,itemList with
  |AsColor ca, AsSize sa -> printfn "%A; %A" ca sa;;

Answer 3

快去小工具数组推导式！

> [| for a in itemList do yield a.Size |];;
val it : int [] = [|1; 2; 3|]
> [| for a in itemList do yield a.Color |];;
val it : string [] = [|"Red"; "Green"; "Blue"|]

您不需要中间的 ItemType 或 ItemEnum 数据结构。