我们说我有一份记录清单,我想通过取中位数来总结一下。更具体地说,我说
data Location = Location { x :: Double, y :: Double }
我有一个测量列表,我想将它总结为中位数Location
,所以类似于:
Location (median (map x measurements)) (median (map y measurements))
这很好,但是如果我有更多嵌套的东西,例如:
data CampusLocation = CampusLocation { firstBuilding :: Location
,secondBuilding :: Location }
我有一个CampusLocation
的列表,我想要一个摘要CampusLocation
,其中中位数递归地应用于所有字段。
在Haskell中最干净的方法是什么?镜头? Uniplate中?
编辑:奖金:
如果不是包含我们想要汇总的字段的记录,我们有一个隐含的列表呢?例如:
data ComplexCampus = ComplexCampus { buildings :: [Location] }
我们如何将[ComplexCampus]
汇总到ComplexCampus
,假设每个buildings
的长度相同?
答案 0 :(得分:5)
以下是summarize :: [ComplexCampus] -> ComplexCampus
的实现,它使用带有Uniplate的镜头(如您所述)将ComplexCampus的列表汇总为单个ComplexCampus。
{-# Language TemplateHaskell,DeriveDataTypeable #-}
import Control.Lens
import Data.Data.Lens
import Data.Typeable
import Data.Data
import Data.List(transpose,genericLength)
data Location = Location { _x :: Double, _y :: Double } deriving(Show,Typeable,Data)
data CampusLocation = CampusLocation { _firstBuilding :: Location, _firsecondBuilding :: Location }deriving(Show,Typeable,Data)
data ComplexCampus = ComplexCampus { _buildings :: [Location] } deriving(Show,Typeable,Data)
makeLenses ''Location
makeLenses ''CampusLocation
makeLenses ''ComplexCampus
l1 = Location 1 10
l2 = Location 2 20
l3 = Location 3 30
c1 = CampusLocation l1 l2
c2 = CampusLocation l2 l3
c3 = CampusLocation l1 l3
campusLocs = [c1,c2,c3]
c1' = ComplexCampus [l1, l2]
c2' = ComplexCampus [l2, l3]
c3' = ComplexCampus [l1, l3]
campusLocs' = [c1',c2',c3']
average l = (sum l) / (genericLength l)
-- returns average location for a list of locations
averageLoc locs = Location {
_x = average $ locs ^.. biplate . x,
_y = average $ locs ^.. biplate . y
}
summarize :: [ComplexCampus] -> ComplexCampus
summarize ccs = ComplexCampus $ ccs ^.. biplate . buildings ^.. folding transpose . to averageLoc
在这里使用biplate可能有点过分,但无论在averageLoc
中,我们都会在地理列表中使用biplate
来获取所有x
字段和所有y
字段。如果您想将ComplexCampus
汇总到一个Location
,我们可以使用biplate
从顶层{x
中提取所有y
值和所有ComplexBuilding
值1}}。
例如:
campusLocs' ^.. biplate . x
为我们提供了所有x值,campusLocs' ^.. biplate . y
为我们提供了所有y值
同样,要获得所有地点,我们可以这样做:
(campusLocs' ^.. biplate) ::[Location]
或者,如果我们想要每一个Double:
(campusLocs' ^.. biplate) ::[Double]