我试图用一个子向量来创建一个向量,该向量由从另一个向量中取出的元素组成,它使用一个子向量索引。b
中的每个元素对应于a
中的元素在放入c
时应该具有的子向量索引。
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]]
我有点卡住了,弄错了
无法匹配stmt列表理解中的预期类型Int与实际类型向量整数k <- b
发布于 2016-05-06 14:23:22
这是行不通的,因为b
是向量,而不是列表:
k <- b
然而,这样做是可行的:
[ ... | k <- toList b ]
接下来,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
不进行排序。
发布于 2016-05-06 15:43:27
在ErikR的帮助下,我想到了以下几点:
let c = fromList [fromList [a ! i | i <- [0..Data.Vector.length b-1], (b ! i)==j] | j <- [0..Data.Vector.maximum(b)]]
它能用,但它不漂亮,更好吗?
发布于 2016-05-06 15:46:25
看起来你想要的可能是
accumulate (flip (:)) (replicate n []) (zip b a)
...although,您必须显式地计算n
,可能是maximum b + 1
。
https://stackoverflow.com/questions/37082155
复制相似问题