我正在尝试创建一个包含子向量的向量，该子向量由使用子向量索引的向量从另一个向量中取出的元素组成。中的每个元素都对应于其中的元素在放入时应具有b的子向量索引。ac
import Data.Vector
let a = fromList [9,2,3,7,4,1,8,5]
let b = fromList [3,3,2,0,1,1,2,2]
let c = fromList [ a ! k | k <- b ]
Expected c = [[7],[4,1],[3,8,5],[9,2]]
我有点卡住了，收到错误
“无法将预期类型 [Int] 与 stmt 列表理解 k <- b 中的实际类型 Vector Integer 匹配”
这不起作用，因为b它是一个向量，而不是一个列表：
k <- b
但是，这可以工作：
[ ... | k <- toList b ]
接下来， and 的类型a是b，Vector Integer并且!运算符采用Int。因此，您需要使用以下方法转换索引fromInteger：
let c = fromList [ a ! fromInteger k | k <- toList b]
更新
这是一种无需重复遍历数组即可执行转换的方法：
import Data.List
fst3 (b,_,_) = b
third (_,_,a) = a
doit :: Vector Int -> Vector Int -> [[Int]]
doit av bv = [ map third g | g <- groups ]
where
triples = zip3 (V.toList bv) [1..] (V.toList av)
groups = groupBy (\s t -> fst3 s == fst3 t) $ sort triples
这基本上是一个Schwartzian 变换，在排序步骤之后添加了 groupBy。三元组的排序以规范的方式完成 - 对第一个坐标进行 lex 排序，然后是第二个坐标，然后是第三个坐标。
还有其他方法可以编写 的表达式groups：
import Data.Funcition (on)
import GHC.Exts (groupWith)
...
groups = groupBy (on (==) fst3) $ sort triples
groups = groupWith fst3 triples
请注意，这groupBy要求对三元组进行排序，groupWith而不需要。
使用列表，这似乎是逻辑
> map (map snd) $ groupBy ((==) `on` fst) $ sortBy (comparing fst) $ zip b a
[[7],[4,1],[3,8,5],[9,2]]
在 ErikR 的帮助下，我想出了这个：
let c = fromList [fromList [a ! i | i <- [0..Data.Vector.length b-1], (b ! i)==j] | j <- [0..Data.Vector.maximum(b)]]
它有效，但它并不漂亮，更好吗？
看起来你想要的可能是
accumulate (flip (:)) (replicate n []) (zip b a)
...尽管您将不得不明确计算n，也许是maximum b + 1。